* adding patch file for torch 1.13

* renaming patch file
This commit is contained in:
Abhishek Jindal 2022-09-20 16:44:36 -07:00 коммит произвёл GitHub
Родитель 357dfd0b11
Коммит 944e6639b8
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
2 изменённых файлов: 14 добавлений и 1 удалений

Просмотреть файл

@ -7,7 +7,7 @@ index 5817449c1a..edc4f7781a 100644
auto* recvbuff = reinterpret_cast<char *>(output.data_ptr());
auto comm = to_nccl_comm(_comm);
-#if defined(USE_ROCM) && ROCM_VERSION >= 50000
+#if 1
+#if 1
NCCL_CHECK(ncclAllToAll(sendbuff , recvbuff , count, type, comm, stream));
#else
NCCL_CHECK(ncclCommCount(comm, &numranks));

Просмотреть файл

@ -0,0 +1,13 @@
diff --git a/torch/csrc/cuda/nccl.cpp b/torch/csrc/cuda/nccl.cpp
index 83729084ae..934bf24ea4 100644
--- a/torch/csrc/cuda/nccl.cpp
+++ b/torch/csrc/cuda/nccl.cpp
@@ -655,7 +655,7 @@ void all2all_single_equal_split(
const auto* sendbuff = reinterpret_cast<char*>(input.data_ptr());
auto* recvbuff = reinterpret_cast<char*>(output.data_ptr());
auto comm = to_nccl_comm(_comm);
-#if defined(USE_ROCM) && ROCM_VERSION >= 50000
+#if 1
NCCL_CHECK(ncclAllToAll(sendbuff, recvbuff, count, type, comm, stream));
#else
NCCL_CHECK(ncclCommCount(comm, &numranks));