pytorch/test/cpp
Ke Wen 4ab852b6c5 [c10d] PGNCCL refactor part 1: adds assert size==1 (#119099)
Breaking #118674 into multiple smaller PRs.
This is the first one.
It adds `assert size==1` to PGNCCL, and refactors some old tests written in multi-device style (which would otherwise fail at the assert).

Pull Request resolved: https://github.com/pytorch/pytorch/pull/119099
Approved by: https://github.com/wconstab
2024-02-06 06:59:47 +00:00
..
aot_inductor [AOTInductor] Add Runtime Constant-folding for AOTInductor (#118765) 2024-02-01 04:54:25 +00:00
api Add torch check for dtype within bilinear (#118900) 2024-02-03 00:02:00 +00:00
c10d [c10d] PGNCCL refactor part 1: adds assert size==1 (#119099) 2024-02-06 06:59:47 +00:00
common
dist_autograd Remove unneeded linking in CMake targets (#109192) 2023-09-15 19:43:25 +00:00
jit Initial torchbind support in PT2 (#117697) 2024-01-19 06:28:20 +00:00
lazy test_lazy: skip HashTest.Scalar (#112747) 2023-11-16 01:22:58 +00:00
lite_interpreter_runtime [2/N] Cleanup header inclusions in torch_cpu by iwyu (#109964) 2023-11-19 20:56:32 +00:00
monitor
profiler [c10] Move profiler clock to libc10 for timestamps (#111972) 2023-10-27 16:18:40 +00:00
rpc Fix typo under test directory (#111304) 2023-10-16 23:06:06 +00:00
tensorexpr [ROCm] remove HCC references (#111975) 2023-10-26 02:39:10 +00:00
__init__.py