pytorch_aot_simple demo error

[yja1]

File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/shark_turbine/aot/exporter.py", line 277, in export
_patch_op_dispatch_for_export()
File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/shark_turbine/aot/decompositions.py", line 78, in _patch_op_dispatch_for_export
import torch._higher_order_ops.auto_functionalize
ModuleNotFoundError: No module named 'torch._higher_order_ops.auto_functionalize'

torch 2.1.2+gitdd11ab3
iree-compiler 20240828.999
iree-runtime 20240828.999
iree-turbine 2.3.0rc20240410