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
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))
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.
Encountered while trying to use PPO algorithm with DirectML.