| .. |
|
codegen
|
|
|
|
data
|
|
|
|
distributed
|
Revert "[PGNCCL] Launch kernel on current stream & remove record_stream entirely (#148590)"
|
2025-03-17 22:43:15 +00:00 |
|
generated
|
|
|
|
opinfo
|
Implement gradient for the residuals of torch.linalg.lstsq (#148526)
|
2025-03-10 12:35:09 +00:00 |
|
optests
|
[opcheck] Improve error reporting; allow atol/rtol overrides (#146488)
|
2025-02-05 21:25:06 +00:00 |
|
test_module
|
|
|
|
__init__.py
|
|
|
|
autocast_test_lists.py
|
|
|
|
autograd_function_db.py
|
|
|
|
check_kernel_launches.py
|
|
|
|
common_cuda.py
|
[BE]: Apply ruff PERF403 to use dict comprehensions more often (#149257)
|
2025-03-18 00:46:07 +00:00 |
|
common_device_type.py
|
cpp_wrapper: use largeTensorTest for test memory checks (#146991)
|
2025-02-27 00:30:21 +00:00 |
|
common_dist_composable.py
|
|
|
|
common_distributed.py
|
Enable FSDP tests on XPU device (#147518)
|
2025-03-04 23:49:37 +00:00 |
|
common_dtype.py
|
|
|
|
common_fsdp.py
|
Enable FSDP tests on XPU device (#147518)
|
2025-03-04 23:49:37 +00:00 |
|
common_jit.py
|
|
|
|
common_methods_invocations.py
|
Implement aten.select.int sharding strategy (#149842)
|
2025-03-27 20:49:00 +00:00 |
|
common_mkldnn.py
|
[BE]: Apply ruff PERF403 to use dict comprehensions more often (#149257)
|
2025-03-18 00:46:07 +00:00 |
|
common_modules.py
|
Remove outdated skipCUDAIfCudnnVersionLessThan decoration (#148940)
|
2025-03-13 18:02:50 +00:00 |
|
common_nn.py
|
|
|
|
common_optimizers.py
|
Remove code for Python < 3.9 (#147097)
|
2025-02-14 03:22:49 +00:00 |
|
common_pruning.py
|
|
|
|
common_quantization.py
|
[BE]: Enable ruff rule SIM113 (#147290)
|
2025-02-16 22:41:16 +00:00 |
|
common_quantized.py
|
wire torch._scaled_mm with fp4 operands to the cublas nvfp4 kernel (#148792)
|
2025-03-27 17:32:20 +00:00 |
|
common_subclass.py
|
|
|
|
common_utils.py
|
Revert "[fbcode]Removing @NoIntBaseDeprecated annotation in caffe2.thrift file (#149742) (#149744)"
|
2025-03-27 22:31:54 +00:00 |
|
composite_compliance.py
|
Revert "[pytree] add APIs to determine a class is a namedtuple or PyStructSequence (#113257)"
|
2025-03-14 23:13:34 +00:00 |
|
custom_op_db.py
|
|
|
|
custom_tensor.py
|
|
|
|
dist_utils.py
|
|
|
|
dynamo_test_failures.py
|
|
|
|
fake_config_module.py
|
|
|
|
fake_config_module2.py
|
|
|
|
fake_config_module3.py
|
|
|
|
hop_db.py
|
Support torch.compile rng selective activation checkpointing with cudagraph (#146878)
|
2025-02-28 00:47:03 +00:00 |
|
hypothesis_utils.py
|
|
|
|
inductor_utils.py
|
Revert "cpp_wrapper: Fix even more tests (#147225)"
|
2025-03-28 17:07:52 +00:00 |
|
jit_metaprogramming_utils.py
|
|
|
|
jit_utils.py
|
Fix linter F821 error (#146665)
|
2025-02-08 07:19:37 +00:00 |
|
logging_tensor.py
|
|
|
|
logging_utils.py
|
Update ruff linter for PEP585 (#147540)
|
2025-02-22 04:45:17 +00:00 |
|
quantization_torch_package_models.py
|
|
|
|
static_module.py
|
|
|
|
subclasses.py
|
[subclass] testing WrapperSubclass respect outer_size, outer_stride (#146897)
|
2025-02-13 15:21:19 +00:00 |
|
torchbind_impls.py
|
Fakify torchbind objects in compile_fx and add tests for SigridTransformsInstanceTorchBind (#149529)
|
2025-03-21 18:58:28 +00:00 |
|
triton_utils.py
|
[Inductor] Use real input to autotune user defined triton kernels (#149553)
|
2025-03-26 16:42:48 +00:00 |
|
two_tensor.py
|
Support subclass constructor capturing in export (#147014)
|
2025-03-16 18:19:19 +00:00 |
