pytorch/test/cpp/c10d
Howard Huang a206e8b027 [small BE] update NcclTest dim size (#101127)
Previously input dimensions are fixed to 3x3, this is a small change to make that configurable. Will be used in future additions to nccl tests
Pull Request resolved: https://github.com/pytorch/pytorch/pull/101127
Approved by: https://github.com/rohan-varma
2023-05-15 23:05:10 +00:00
..
example
CMakeLists.txt
CUDATest.cu
CUDATest.hpp
FileStoreTest.cpp
HashStoreTest.cpp
ProcessGroupGlooAsyncTest.cpp
ProcessGroupGlooTest.cpp
ProcessGroupMPITest.cpp
ProcessGroupNCCLErrorsTest.cpp
ProcessGroupNCCLTest.cpp
ProcessGroupUCCTest.cpp
StoreTestCommon.hpp
TCPStoreTest.cpp
TestUtils.hpp