search

microsoft / DirectML

DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for common machine learning tasks across a broad range of supported hardware and drivers, including all DirectX 12-capable GPUs from vendors such as AMD, Intel, NVIDIA, and Qualcomm.
MIT License
2.18k stars 288 forks source link

Could not run 'aten::reflection_pad1d' #294

Open qyw233 opened 2 years ago

qyw233 commented 2 years ago

I try to run https://[github.com](https://github.com/bytedance/GiantMIDI-Piano)/bytedance/GiantMIDI-Piano with pytorch-ml， after changing cuda to dml，I found an error: ...... File "/home/john/miniconda3/envs/pydml/lib/python3.8/site-packages/torch/nn/functional.py", line 4003, in _pad return torch._C._nn.reflection_pad1d(input, pad) RuntimeError: Could not run 'aten::reflection_pad1d' with arguments from the 'DML' backend. This could be because the operator doesn't exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit https://fburl.com/ptmfixes for possible resolutions. 'aten::reflection_pad1d' is only available for these backends: [CPU, QuantizedCPU, BackendSelect, Named, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradNestedTensor, UNKNOWN_TENSOR_TYPE_ID, AutogradPrivateUse1, AutogradPrivateUse2, AutogradPrivateUse3, Tracer, Autocast, Batched, VmapMode].

Does not pytorch-ml support aten::reflection_pad1d now?

NicerWang commented 1 year ago

Yes All the supported operators: https://github.com/microsoft/DirectML/wiki/PyTorch-DirectML-Operator-Roadmap

Adele101 commented 1 year ago

Hi @qyw233, thanks for the request. We will add aten::reflection_pad1d to our backlog for operator support.