To reproduce: upgrade to torch 2.2.0+ run tests.py
To reproduce:
upgrade to torch 2.2.0+
run tests.py