microsoft/DirectML

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

Djdefrag opened this issue · 4 comments

Djdefrag commented

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 commented

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

Djdefrag commented

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

  • 👍1
Djdefrag commented

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 commented

Up. We really need better DirectML support in PyTorch.

  • 👍1