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.22k stars 298 forks source link

Operator 'aten::binary_cross_entropy' is not currently supported #411

Open rosaxy opened 1 year ago

rosaxy commented 1 year ago

I got this message while trying pytorch:

UserWarning: The operator 'aten::binary_cross_entropy' is not currently supported on the DML backend and will fall back to run on the CPU.

Looking forward to its implementation :)

Adele101 commented 1 year ago

Hi @rosaxy, 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.

steking888 commented 7 months ago

no ! i need answer

iperov commented 3 months ago

one of the main operators is still not implemented. Unable to train GAN without it.

What a shame.