pytorch/test/cpp
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
..
aot_inductor [inductor] Fix AOTInductor (#99203) 2023-04-25 14:42:12 +00:00
api Re enable AutogradNotImplementedFallback on Windows (#101062) 2023-05-15 13:41:06 +00:00
c10d [small BE] update NcclTest dim size (#101127) 2023-05-15 23:05:10 +00:00
common
dist_autograd
jit [IValue] Better handle sparseTensors in extractStorages (#100783) 2023-05-11 23:44:51 +00:00
lazy std/var: support floating point correction value (#94073) 2023-02-23 05:50:45 +00:00
lite_interpreter_runtime
monitor
profiler
rpc
tensorexpr Revert "Added round_with_scale_factor arg to ATen (#97868)" 2023-04-28 20:47:00 +00:00
__init__.py