Issue-Report: PyTroch v. 0.2.1.dev240521 - Dropout no longer working

[mor-lu]

Since release 0.2.1.dev240521 when using a dropout layer the following error displays and the whole model is trained on the cpu:

'aten:native.dropout' is not currently supported on the DML backend and will fall back to run on the CPU.

Reverting to the previous version 0.2.0.dev230426 fixed the issue.

Windows 11
AMD Radeon rx7600 GPU

[joshjkim]

Hi @mor-lu , thank you for the feedback. We've now added support for native_dropout in our latest torch-directml 0.2.2.dev240614 release. Please run pip install torch-directml --upgrade to update to this version