| .. |
|
codegen
|
|
|
|
data
|
|
|
|
distributed
|
[BE] move init_multigpu_helper to common_distributed (#67050)
|
2021-10-22 17:16:11 -07:00 |
|
generated
|
|
|
|
test_module
|
Drop unused imports (#49972)
|
2021-01-13 12:26:17 -08:00 |
|
__init__.py
|
|
|
|
autocast_test_lists.py
|
add operation list for AutocastCPU (#63534)
|
2021-08-30 19:30:33 -07:00 |
|
check_kernel_launches.py
|
move kernel launch checks from torch.testing to torch.testing._internal.check_kernel_launches (#60862)
|
2021-09-29 00:39:22 -07:00 |
|
common_cuda.py
|
Add cuSPARSE descriptors and update CSR addmm (#60838)
|
2021-09-30 11:32:51 -07:00 |
|
common_device_type.py
|
Enable warning tests for nondeterministic backward functions (#66736)
|
2021-10-21 12:51:53 -07:00 |
|
common_distributed.py
|
[BE] move init_multigpu_helper to common_distributed (#67050)
|
2021-10-22 17:16:11 -07:00 |
|
common_dtype.py
|
deprecate device getter from torch.testing namespace (#63844)
|
2021-09-29 02:40:52 -07:00 |
|
common_fsdp.py
|
Add more FSDP unit tests to cover core logic, freezing weights and flatten parameter wrapper (#66904)
|
2021-10-22 16:50:52 -07:00 |
|
common_fx2trt.py
|
Add common_fx2trt.py (#66579)
|
2021-10-13 21:24:11 -07:00 |
|
common_jit.py
|
Max Pool with indices (#64121)
|
2021-09-15 13:45:46 -07:00 |
|
common_methods_invocations.py
|
Improve assert failure message in test_get_torch_func_signature_exhaustive (#67039)
|
2021-10-25 14:20:38 -07:00 |
|
common_modules.py
|
TST Adds test for non-contiguous tensors (#64954)
|
2021-09-24 15:05:09 -07:00 |
|
common_nn.py
|
Do rowwisemoments computation in float for half LayerNorm (#66920)
|
2021-10-22 09:50:42 -07:00 |
|
common_quantization.py
|
[quant][embedding qat] Add eager QAT test for EmbeddingBag+Linear model (#66334)
|
2021-10-19 07:03:36 -07:00 |
|
common_quantized.py
|
[quant][refactor tests] split test_workflow_module into test_workflow_ops and test_workflow_module (#58963)
|
2021-05-26 23:01:08 -07:00 |
|
common_utils.py
|
Do not build PyTorch with caffe2 by default (#66658)
|
2021-10-21 20:32:47 -07:00 |
|
dist_utils.py
|
Use higher timeout for TSAN tests. (#65391)
|
2021-09-21 12:08:27 -07:00 |
|
hypothesis_utils.py
|
|
|
|
jit_metaprogramming_utils.py
|
Add relative and absolute tolerances for matrix_rank, pinv (#63102)
|
2021-10-17 22:15:42 -07:00 |
|
jit_utils.py
|
Revert "Revert D30752939: [pytorch][PR] nvfuser update" (#65137)
|
2021-09-22 04:54:51 -07:00 |
|
opinfo_helper.py
|
deprecate dtype getters from torch.testing namespace (#63554)
|
2021-09-07 08:58:51 -07:00 |
|
quantization_torch_package_models.py
|
fx quant: add a BC test for loading old torch.package models (#65538)
|
2021-10-11 08:23:38 -07:00 |
333717eaf0