Moco NotImplementedError: xpu not supported #489

🐛 Describe the bug

torchbench_amp_fp16_training
xpu train moco
Traceback (most recent call last):
File "/home/sdp/actions-runner/_work/torch-xpu-ops/pytorch/benchmarks/dynamo/common.py", line 4177, in run
) = runner.load_model(
File "/home/sdp/actions-runner/_work/torch-xpu-ops/pytorch/benchmarks/dynamo/torchbench.py", line 320, in load_model
benchmark = benchmark_cls(
File "/home/sdp/actions-runner/_work/torch-xpu-ops/benchmark/torchbenchmark/util/model.py", line 39, in call
obj = type.call(cls, *args, **kwargs)
File "/home/sdp/actions-runner/_work/torch-xpu-ops/benchmark/torchbenchmark/models/moco/init.py", line 80, in init
raise NotImplementedError(f"{device} not supported")
NotImplementedError: xpu not supported

model_fail_to_load

Versions

torch-xpu-ops: 31c4001
pytorch: 0f81473 + PRs: 127277,129120
device: PVC 1100, 803.61, 0.5.1