search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.47k stars 471 forks source link

Model support for `pytorch_unet` with Torch_XLA2 #8169

Closed ManfeiBai closed 1 week ago

ManfeiBai commented 1 week ago

Fix the model test for pytorch_unet.py

  1. setup env according to Run a model under torch_xla2
  2. Run model test under run_torchbench/ with python models/your_target_model_name.py
  3. Fix the failure.

Please refer to this guide as guide to fix:

Also refer to these PRs:

ManfeiBai commented 1 week ago

passed locally