BasicVSR++运行device{NPU：0}时报错

RuntimeError: modulated_deformable_im2col_impl: implementation for device xla:0 not found.


model = NPUDataParallel(model.npu(), dim=0, device_ids=range(cfg.gpus))

xla这个错误是走到了mmlab的自定义算子库


```

Traceback (most recent call last):

File "train.py", line 174, in <module>

main()

File "train.py", line 170, in main

meta=meta)

File "/usr/local/python3.7.5/lib/python3.7/site-packages/mmedit-0.16.0-py3.7.egg/mmedit/apis/train.py", line 115, in train_model

File "/usr/local/python3.7.5/lib/python3.7/site-packages/mmedit-0.16.0-py3.7.egg/mmedit/apis/train.py", line 370, in _non_dist_train

runner.run(data_loaders, cfg.workflow, cfg.total_iters)

File "/home/test_user07/mmcv/mmcv/runner/iter_based_runner.py", line 144, in run

iter_runner(iter_loaders[i], **kwargs)

File "/home/test_user07/mmcv/mmcv/runner/iter_based_runner.py", line 64, in train

outputs = self.model.train_step(data_batch, self.optimizer, **kwargs)

File "/home/test_user07/mmcv/mmcv/parallel/data_parallel.py", line 77, in train_step

return self.module.train_step(*inputs[0], **kwargs[0])

File "/usr/local/python3.7.5/lib/python3.7/site-packages/mmedit-0.16.0-py3.7.egg/mmedit/models/restorers/basicvsr.py", line 103, in train_step

outputs = self(**data_batch, test_mode=False)

File "/home/test_user07/.local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl

result = self.forward(*input, **kwargs)

File "/home/test_user07/mmcv/mmcv/runner/fp16_utils.py", line 119, in new_func

return old_func(*args, **kwargs)

File "/usr/local/python3.7.5/lib/python3.7/site-packages/mmedit-0.16.0-py3.7.egg/mmedit/models/restorers/basic_restorer.py", line 77, in forward

return self.forward_train(lq, gt)

File "/usr/local/python3.7.5/lib/python3.7/site-packages/mmedit-0.16.0-py3.7.egg/mmedit/models/restorers/basic_restorer.py", line 90, in forward_train

output = self.generator(lq)

File "/home/test_user07/.local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl

result = self.forward(*input, **kwargs)

File "/usr/local/python3.7.5/lib/python3.7/site-packages/mmedit-0.16.0-py3.7.egg/mmedit/models/backbones/sr_backbones/basicvsr_pp.py", line 347, in forward

feats = self.propagate(feats, flows, module)

File "/usr/local/python3.7.5/lib/python3.7/site-packages/mmedit-0.16.0-py3.7.egg/mmedit/models/backbones/sr_backbones/basicvsr_pp.py", line 218, in propagate

flow_n1, flow_n2)

File "/home/test_user07/.local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl

result = self.forward(*input, **kwargs)

File "/usr/local/python3.7.5/lib/python3.7/site-packages/mmedit-0.16.0-py3.7.egg/mmedit/models/backbones/sr_backbones/basicvsr_pp.py", line 433, in forward

self.deform_groups)

File "/home/test_user07/mmcv/mmcv/ops/modulated_deform_conv.py", line 95, in forward

with_bias=ctx.with_bias)

RuntimeError: modulated_deformable_im2col_impl: implementation for device xla:0 not found.

THPModule_npu_shutdown success.

```