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.17k stars 288 forks source link

operator 'aten::normal.Tensor_Tensor' is not currently supported on the DML backend #405

Open Spidper opened 1 year ago

Spidper commented 1 year ago

Encountered while trying to use PPO algorithm with DirectML.

E:\miniconda3\envs\drl\lib\site-packages\torch\distributions\normal.py:70: UserWarning: The operator 'aten::normal.Tensor_Tensor' is not currently supported on the 
DML backend and will fall back to run on the CPU. This may have performance implications. (Triggered internally at D:\a\_work\1\s\pytorch-directml-plugin\torch_directml\csrc\dml\dml_cpu_fallback.cpp:17.)
  return torch.normal(self.loc.expand(shape), self.scale.expand(shape))
Adele101 commented 11 months ago

Hi @Spidper, thank you for submitting this issue. While I can't provide a timeline for resolution as the moment, please know that your feedback is valuable to us. We will follow up once we can review this issue.