Unable to Import TorchDispatch

autograd
1

Hi, I’m unable to figure out how to import torchdispatch for PyTorch v1.12.0.
The same import statement works without error for Pytorch2.0, but I needed to use it for Pytorch v1.12.0

I’m getting the below errror :

image
image1832×120 7.44 KB

The import statement looks like below :
image

2

I do see TorchDispatchMode show up here pytorch/_python_dispatch.py at release/1.12 · pytorch/pytorch · GitHub