Pytorch mainline support and newer directml version for torch-directml?

[Djdefrag]

Hi,

there is some roadmap for newer torch-directml release?

Latest working version of torch-directml is 0.1.13.1.dev230413 that is built with:

• torch 1.13.1 released in December 2022 (latest 2.1.2 released last week)
• and directml 1.10.1 release in January 2023 (latest 1.13 released last week)

so we are missing almost one year of torch and directml development and bugifx :(

[sorasoras]

I think the current one is link to torch 2.0 but still, that's not 2.1.2 mainline.

[Djdefrag]

Hi, yes. Latest version is 0.2.0.dev230426 based on torch 2.0 but has some bugs for me

Error: [F D:\a_work\1\s\pytorch-directml\caffe2\dml\dml_tensor_desc.cc:81] Check failed: rank <= DML_TENSOR_DIMENSION_COUNT_MAX

[Djdefrag]

Update 31/01/2024

a new version of pytorch has been released, actually release are:

• torch 2.2.0 (actually 1.13.1)
• directml 1.13.1 (actually 1.10.1)

so we are behind 6 versions of pytorch and 5 versions of directml :(

[deiteris]

Up. We really need better DirectML support in PyTorch.