frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
ad67170c8b9c7ddd3442e32369cb9a0be7631d91
pytorch/test
History
Isalia20 ad67170c8b [MPS] sparse matmuls (#165232) 
Implements matmuls for sparse tensors. With this commit most of the core sparse operations should be implemented. Fixes:
https://github.com/pytorch/pytorch/issues/156540
https://github.com/pytorch/pytorch/issues/129842

Should be merged after:
https://github.com/pytorch/pytorch/pull/165102

To compare MPS and CPU, you can use this script:
```python
import torch
import time
import matplotlib.pyplot as plt

B, I, J, K = 8, 20000, 20000, 20000
num_iterations = 500

nnz_values = [10, 50, 100, 200, 500, 1000, 2000, 5000, 10000, 20000, 100000]
speedups = []

for nnz in nnz_values:
    indices = torch.stack([
        torch.randint(0, B, (nnz,)),
        torch.randint(0, I, (nnz,)),
        torch.randint(0, J, (nnz,)),
    ])
    values = torch.rand(nnz)

    sparse = torch.sparse_coo_tensor(indices, values, size=(B, I, J), device="mps").coalesce()
    dense = torch.randn(B, J, 200, device="mps")

    t1 = time.time()
    for _ in range(num_iterations):
        result = torch.bmm(sparse, dense)
    torch.mps.synchronize()
    t2 = time.time()
    mps_time = (t2 - t1) / num_iterations

    sparse_cpu = sparse.cpu()
    dense_cpu = dense.cpu()
    t1 = time.time()
    for _ in range(num_iterations):
        result_cpu = torch.bmm(sparse_cpu, dense_cpu)
    t2 = time.time()
    cpu_time = (t2 - t1) / num_iterations

    speedup = cpu_time / mps_time
    speedups.append(speedup)
    print(f"nnz={nnz}: MPS={mps_time:.6f}s, CPU={cpu_time:.6f}s, Speedup={speedup:.2f}x")

plt.figure(figsize=(10, 6))
plt.plot(nnz_values, speedups, marker='o', linewidth=2, markersize=8)
plt.xlabel('Number of Non-Zero Elements (nnz)', fontsize=12)
plt.ylabel('Speedup (CPU time / MPS time)', fontsize=12)
plt.title('MPS vs CPU Speedup for Sparse-Dense BMM', fontsize=14)
plt.grid(True, alpha=0.3)
plt.axhline(y=1, color='r', linestyle='--', alpha=0.5)
plt.xscale('log')
plt.tight_layout()
plt.show()
```

## Tested on M1 Pro
<img width="1000" height="600" alt="Figure_1" src="https://github.com/user-attachments/assets/4a2402ec-3dc4-402d-8196-a0426906ca3d" />

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165232
Approved by: https://github.com/malfet
2025-10-18 09:04:42 +00:00
..
ao/sparsity
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
autograd
Fix unused Python variables in test/[a-d]* (#134665)
2024-12-13 22:13:12 +00:00
backends/xeon
…
benchmark_utils
[BE][3/6] fix typos in test/ (#157637)
2025-07-17 12:08:33 +00:00
compiled_autograd_skips
[ca] skip on some PYTORCH_TEST_WITH_DYNAMO=1 autograd tests (#156374)
2025-06-21 18:33:38 +00:00
cpp
Refactor out headeronly ArrayRef (#164991)
2025-10-17 18:32:39 +00:00
cpp_api_parity
[BE][PYFMT] migrate PYFMT for test/[a-h]*/ to ruff format (#144555)
2025-06-24 04:53:54 +00:00
cpp_extensions
Widen ops support to take in IntHOArrayRef vs only std::vec (#165152)
2025-10-17 18:32:39 +00:00
custom_backend
[Cmake] Make PyTorch buildable by CMake-4.x (#150203)
2025-03-29 01:39:13 +00:00
custom_operator
Using the latest torch.library.register_fake API instead of torch.library.impl_abstract (#158839)
2025-07-25 02:37:30 +00:00
distributed
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
distributions
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
dynamo
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
dynamo_expected_failures
Patch the flex_attention._get_mod_type to not use inspect.signature when computing num_positional_args (an alternative fix for flex attention graph break on create_block_mask) (#164923)
2025-10-17 17:44:45 +00:00
dynamo_skips
Add functions to setup PrivateUse1 as a python backend device. (#157859)
2025-10-01 21:32:59 +00:00
error_messages
…
expect
Add scaled_grouped_mm_v2 and python API (#165154)
2025-10-15 17:47:23 +00:00
export
Revert "Update gm.print_readable to include Annotation (#165397)"
2025-10-17 22:35:50 +00:00
forward_backward_compatibility
Revert "[BC-Breaking] Remove long-deprecated casting functions from native_functions.yaml (#164641)"
2025-10-09 15:42:51 +00:00
functorch
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
fx
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
higher_order_ops
Revert "Update gm.print_readable to include Annotation (#165397)"
2025-10-17 22:35:50 +00:00
inductor
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
inductor_expected_failures
[dynamo] Graph break on on user-defined class in compiled region (#161670)
2025-09-10 04:39:20 +00:00
inductor_skips
[BE] Remove test_ops from FIXME_inductor_dont_reset_dynamo (#145307)
2025-01-27 18:12:39 +00:00
jit
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
jit_hooks
[Cmake] Make PyTorch buildable by CMake-4.x (#150203)
2025-03-29 01:39:13 +00:00
lazy
Do not decompose in functionalization/proxy tensor if autograd wouldn't have decomposed (#164939)
2025-10-11 01:03:55 +00:00
mobile
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
nn
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
onnx
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
optim
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
package
[Torch Package] Make get names of OrderedImporters support fallback to importers (#155743)
2025-08-06 02:26:10 +00:00
profiler
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
quantization
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
scripts
[BE][PYFMT] migrate PYFMT for test/[i-z]*/ to ruff format (#144556)
2025-07-29 03:26:09 +00:00
strobelight/examples
Enable strobelight profiling specific compile frame ids using COMPILE_STROBELIGHT_FRAME_FILTER (#147549)
2025-02-22 03:44:53 +00:00
test_img
…
torch_np
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
typing
Revert "Add __init__.pyi to torch/linalg (#160750)"
2025-09-02 16:53:55 +00:00
xpu
[WOQ] Add XPU kernel for _weight_int8pack_mm (#160938)
2025-09-19 07:37:14 +00:00
_test_bazel.py
…
allowlist_for_publicAPI.json
Revert "[BC-Breaking] Remove long-deprecated casting functions from native_functions.yaml (#164641)"
2025-10-09 15:42:51 +00:00
bench_mps_ops.py
[BE] Remove macos-13 guard from bench_mps_ops (#159732)
2025-08-03 20:53:58 +00:00
conftest.py
Stop parsing command line arguments every time common_utils is imported. (#156703)
2025-10-02 22:22:04 +00:00
create_dummy_torchscript_model.py
…
HowToWriteTestsUsingFileCheck.md
…
linear.py
…
load_torchscript_model.py
Flip default on weights_only (#137602)
2024-11-04 18:30:29 +00:00
minioptest_failures_dict.json
…
mkl_verbose.py
…
mkldnn_verbose.py
…
pytest_shard_custom.py
…
run_doctests.sh
…
run_test.py
[CI] Fix doctest job if build without distributed (#165449)
2025-10-14 19:19:03 +00:00
simulate_nccl_errors.py
[BE]: Update ruff to 0.11.8 (#153249)
2025-05-12 18:30:52 +00:00
slow_tests.json
Update slow tests (#165301)
2025-10-13 11:47:32 +00:00
test_accelerator.py
Add UT for torch.accelerator memory-related API (#155200)
2025-08-08 17:41:22 +00:00
test_ao_sparsity.py
[testing] Add test owner labels for some ao sparse tests (#163203)
2025-09-18 16:08:13 +00:00
test_appending_byte_serializer.py
Check integrity of bytes in AppendingByteSerializer (#152139)
2025-04-26 18:10:58 +00:00
test_autocast.py
Revert "158232 Fix autocast cache incorrectly retaining no_grad state (#165068)"
2025-10-16 23:08:27 +00:00
test_autograd_fallback.py
Fix TestAutogradFallback flaky tests under Dynamo: migrate to lib._destroy() (#159443)
2025-07-30 19:30:55 +00:00
test_autograd.py
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
test_autoload.py
…
test_binary_ufuncs.py
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
test_bundled_images.py
Add __main__ guards to tests (#154716)
2025-06-04 14:38:13 +00:00
test_bundled_inputs.py
Pyrefly suppressions 7/n (#164913)
2025-10-08 07:27:17 +00:00
test_ci_sanity_check_fail.py
…
test_comparison_utils.py
[export] Add device and dtype fields to assert_tensor_metadata (#141071)
2024-11-22 20:54:55 +00:00
test_compile_benchmark_util.py
…
test_complex.py
Pyrefly suppressions 7/n (#164913)
2025-10-08 07:27:17 +00:00
test_content_store.py
[ROCm] Enabling several UTs (#161715)
2025-09-09 15:49:21 +00:00
test_cpp_api_parity.py
Revert "Stop parsing command line arguments every time common_utils is imported. (#156703)"
2025-08-04 20:37:39 +00:00
test_cpp_extensions_aot.py
Replace setup.py install with pip install (#156711)
2025-09-29 15:15:10 +00:00
test_cpp_extensions_jit.py
[MPSHooks] Release pending command encoder (#164093)
2025-09-29 17:50:12 +00:00
test_cpp_extensions_mtia_backend.py
Revert "Generalize poison fork logic for each device backend (#144664)"
2025-04-10 21:02:14 +00:00
test_cpp_extensions_stream_and_event.py
[Easy] Add more check for elapsedTime of torch.xxx.Event and torch.Event (#151404)
2025-04-25 20:15:04 +00:00
test_cuda_expandable_segments.py
Revert "Use absolute path path.resolve() -> path.absolute() (#129409)"
2025-01-04 14:17:20 +00:00
test_cuda_multigpu.py
[ROCm] Enabling several UTs (#161715)
2025-09-09 15:49:21 +00:00
test_cuda_nvml_based_avail.py
More ruff SIM fixes (#164695)
2025-10-09 03:24:50 +00:00
test_cuda_primary_ctx.py
[ROCm][CI] skip TestCudaPrimaryCtx.test_set_device_0 (#163693)
2025-09-23 22:15:10 +00:00
test_cuda_sanitizer.py
[4/N] Apply py39 ruff and pyupgrade fixes (#143257)
2025-01-04 10:47:51 +00:00
test_cuda_trace.py
Fix unused Python variables in test/[e-z]* (#136964)
2024-12-18 23:02:30 +00:00
test_cuda.py
[Mem Snapshot] Add Metadata Field (#165490)
2025-10-17 23:46:02 +00:00
test_custom_ops.py
[Bugfix][vLLM] Explicitly do not support instead of crashing for named tuples in infer schema (#165191)
2025-10-14 14:18:42 +00:00
test_dataloader.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_datapipe.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_decomp.py
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
test_determination.py
…
test_dispatch.py
Typo fixes for "overridden" in comments and function names (#155944)
2025-06-14 03:37:38 +00:00
test_dlpack.py
[Fix] Completely remove stride normalization on DLPack Tensor (#164161)
2025-10-14 17:17:11 +00:00
test_dynamic_shapes.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_expanded_weights.py
Revert "Stop parsing command line arguments every time common_utils is imported. (#156703)"
2025-08-04 20:37:39 +00:00
test_extension_utils.py
Move privateuse1 test out of test_utils and make them serial (#145380)
2025-01-23 00:31:39 +00:00
test_fake_tensor.py
[FakeTensor] Supplement the relevant logic for converting conv1d to conv2d in meta_conv (#160408)
2025-09-26 15:45:02 +00:00
test_file_check.py
Fix unused Python variables in test/[e-z]* (#136964)
2024-12-18 23:02:30 +00:00
test_flop_counter.py
[ROCm] Enable MI355 CI on PRs, and run full set of UTs on PRs (#160215)
2025-10-09 18:03:12 +00:00
test_foreach.py
Remove old ROCm version check in tests (#164245)
2025-10-06 22:42:01 +00:00
test_function_schema.py
Fix unused Python variables in test/[e-z]* (#136964)
2024-12-18 23:02:30 +00:00
test_functional_autograd_benchmark.py
Enable Windows tests (#146666)
2025-02-08 00:55:20 +00:00
test_functional_optim.py
[4/N] Apply py39 ruff and pyupgrade fixes (#143257)
2025-01-04 10:47:51 +00:00
test_functionalization_of_rng_ops.py
[BE][2/6] fix typos in test/ (test/test_*.py) (#157636)
2025-07-09 11:02:23 +00:00
test_functionalization.py
[BE][2/6] fix typos in test/ (test/test_*.py) (#157636)
2025-07-09 11:02:23 +00:00
test_futures.py
…
test_fx_experimental.py
[8/N] Apply ruff UP035 rule (#165214)
2025-10-15 03:18:57 +00:00
test_fx_passes.py
[BE][2/6] fix typos in test/ (test/test_*.py) (#157636)
2025-07-09 11:02:23 +00:00
test_fx_reinplace_pass.py
[dynamic shapes] avoid unnecessary slices (#157528)
2025-07-10 06:34:46 +00:00
test_fx.py
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
test_hop_infra.py
Support torch.compile rng selective activation checkpointing with cudagraph (#146878)
2025-02-28 00:47:03 +00:00
test_hub.py
Add __main__ guards to tests (#154716)
2025-06-04 14:38:13 +00:00
test_import_stats.py
…
test_indexing.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_itt.py
…
test_jit_autocast.py
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
test_jit_disabled.py
…
test_jit_fuser_legacy.py
Stop parsing command line arguments every time common_utils is imported. (#156703)
2025-10-02 22:22:04 +00:00
test_jit_fuser_te.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_jit_fuser.py
Stop parsing command line arguments every time common_utils is imported. (#156703)
2025-10-02 22:22:04 +00:00
test_jit_legacy.py
Stop parsing command line arguments every time common_utils is imported. (#156703)
2025-10-02 22:22:04 +00:00
test_jit_llga_fuser.py
Fix unused Python variables in test/[e-z]* (#136964)
2024-12-18 23:02:30 +00:00
test_jit_profiling.py
…
test_jit_simple.py
…
test_jit_string.py
More ruff SIM fixes (#164695)
2025-10-09 03:24:50 +00:00
test_jit.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_jiterator.py
Fix unused Python variables in test/[e-z]* (#136964)
2024-12-18 23:02:30 +00:00
test_kernel_launch_checks.py
[testing] Add test owner labels for some cuda? tests (#163296)
2025-09-26 18:26:56 +00:00
test_legacy_vmap.py
[BE][2/6] fix typos in test/ (test/test_*.py) (#157636)
2025-07-09 11:02:23 +00:00
test_license.py
[BE][2/6] fix typos in test/ (test/test_*.py) (#157636)
2025-07-09 11:02:23 +00:00
test_linalg.py
[ROCm][tunableop] Improvements to tunableop Numerical Check (#163079)
2025-10-15 22:26:47 +00:00
test_logging.py
…
test_masked.py
[BE][2/6] fix typos in test/ (test/test_*.py) (#157636)
2025-07-09 11:02:23 +00:00
test_maskedtensor.py
Fix MaskedTensor to device ignored mask (#151205)
2025-07-21 21:44:49 +00:00
test_matmul_cuda.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_meta.py
[Inductor] support mixed dtype in the native_layer_norm_backward meta function (#159830)
2025-09-17 20:29:12 +00:00
test_metal.py
Fix unused Python variables in test/[e-z]* (#136964)
2024-12-18 23:02:30 +00:00
test_mkl_verbose.py
…
test_mkldnn_fusion.py
[4/N] Apply py39 ruff and pyupgrade fixes (#143257)
2025-01-04 10:47:51 +00:00
test_mkldnn_verbose.py
…
test_mkldnn.py
Enable TF32 as fp32 internal precision for matmul/linear/conv (#157520)
2025-07-17 08:57:34 +00:00
test_mobile_optimizer.py
Fix unused Python variables in test/[e-z]* (#136964)
2024-12-18 23:02:30 +00:00
test_model_exports_to_core_aten.py
Replace export_for_training with export (#162396)
2025-09-10 14:19:34 +00:00
test_module_tracker.py
Fix unused Python variables in test/[e-z]* (#136964)
2024-12-18 23:02:30 +00:00
test_modules.py
[ROCm][CI] unskip some test_memory_format tests (#162766)
2025-09-12 15:02:40 +00:00
test_monitor.py
Fix unused Python variables in test/[e-z]* (#136964)
2024-12-18 23:02:30 +00:00
test_mps.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_multiprocessing_spawn.py
Change forkserver test to only run below 3.13.8 (#165667)
2025-10-16 19:34:10 +00:00
test_multiprocessing.py
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
test_namedtensor.py
Fix tensor creation with empty names crash (#163957)
2025-09-26 17:41:00 +00:00
test_namedtuple_return_api.py
…
test_native_functions.py
[4/N] Apply py39 ruff and pyupgrade fixes (#143257)
2025-01-04 10:47:51 +00:00
test_native_mha.py
[ROCm] Update to AOTriton 0.8b (#140172)
2024-12-06 21:45:18 +00:00
test_nestedtensor.py
[NJT] Fix schema validation error in jagged functions (#165307)
2025-10-13 17:59:18 +00:00
test_nn.py
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
test_nnapi.py
[4/N] Apply py39 ruff and pyupgrade fixes (#143257)
2025-01-04 10:47:51 +00:00
test_numa_binding.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_numba_integration.py
[testing] Add test owner labels for some cuda? tests (#163296)
2025-09-26 18:26:56 +00:00
test_numpy_interop.py
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
test_opaque_obj_v2.py
[opaque_obj_v2] PyObject custom op schema type (#165004)
2025-10-14 20:21:04 +00:00
test_opaque_obj.py
[opaque_obj] Add make_fx tracing support (#163278)
2025-10-08 09:09:16 +00:00
test_openmp.py
Fix unused Python variables in test/[e-z]* (#136964)
2024-12-18 23:02:30 +00:00
test_ops_fwd_gradients.py
…
test_ops_gradients.py
Enable more nightly tests on s390x (#148452)
2025-03-18 16:09:39 +00:00
test_ops_jit.py
[BE][2/6] fix typos in test/ (test/test_*.py) (#157636)
2025-07-09 11:02:23 +00:00
test_ops.py
[1/N]Enable some tests in test_ops.TestCommon on Intel GPU (#159944)
2025-09-29 09:08:04 +00:00
test_optim.py
Fix LBFGS wolfe max iteration (#161488)
2025-09-16 12:07:50 +00:00
test_out_dtype_op.py
[Codemod][AddExplicitStrictExportArg] caffe2/test (#143688)
2024-12-27 07:58:44 +00:00
test_overrides.py
Revert "[Reland] Use std::string_view in torchgen (#158625)"
2025-09-13 07:52:50 +00:00
test_package.py
…
test_per_overload_api.py
[BE][2/6] fix typos in test/ (test/test_*.py) (#157636)
2025-07-09 11:02:23 +00:00
test_prims.py
[Bugfix] Match eager stride semantics for cloned tensors with preserve_format in compile (#163017)
2025-09-19 19:41:33 +00:00
test_privateuseone_python_backend.py
Add functions to setup PrivateUse1 as a python backend device. (#157859)
2025-10-01 21:32:59 +00:00
test_proxy_tensor.py
Fix refine_ranges corner case (#164075) (#164846)
2025-10-08 18:42:37 +00:00
test_pruning_op.py
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
test_public_bindings.py
Revert "[CUDA] Add experimental green context support for SM carveout (#159104)"
2025-10-07 20:51:22 +00:00
test_python_dispatch.py
Fix double dispatch to Python for detach (#163671)
2025-10-15 17:24:50 +00:00
test_pytree.py
[BE][pytree] cleanup parameterized pytree tests (#160842)
2025-09-05 20:15:29 +00:00
test_quantization.py
Remove pytorch quant docs since we are moving to torchao (#157766)
2025-07-11 03:21:47 +00:00
test_reductions.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_rename_privateuse1_to_existing_device.py
[Device] Add support for PrivateUse1 device type in parse_type function (#157609)
2025-07-17 01:27:44 +00:00
test_scaled_matmul_cuda.py
Fix B200 test fails in scaled_mm (#165747)
2025-10-17 17:52:19 +00:00
test_scatter_gather_ops.py
fix deterministic scatter_add path for multi-d tensors (#162866)
2025-09-15 06:50:00 +00:00
test_schema_check.py
[inductor] slow test some Windows UTs. (#160267)
2025-08-10 18:35:42 +00:00
test_segment_reductions.py
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
test_serialization.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_set_default_mobile_cpu_allocator.py
…
test_shape_ops.py
[Quant] flip: throw runtime error for QUInt4x2 and QUInt2x4 input (#147430)
2025-02-25 03:47:40 +00:00
test_show_pickle.py
…
test_sort_and_select.py
Add dtype checks in meta dispatch for various ordering ops (#159556)
2025-08-14 17:06:27 +00:00
test_sparse_csr.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_sparse_semi_structured.py
Remove test conditions for CUDA<12 (#163495)
2025-09-23 07:52:00 +00:00
test_sparse.py
[MPS] sparse matmuls (#165232)
2025-10-18 09:04:42 +00:00
test_spectral_ops.py
fix numpy compatibility for 2d small list indices (#154806)
2025-06-04 01:58:52 +00:00
test_stateless.py
[BE][2/6] fix typos in test/ (test/test_*.py) (#157636)
2025-07-09 11:02:23 +00:00
test_static_runtime.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_subclass.py
Fix unused Python variables in test/[e-z]* (#136964)
2024-12-18 23:02:30 +00:00
test_sympy_utils.py
[1/N] Apply UP035 rule in tests (#163947)
2025-09-29 01:42:01 +00:00
test_tensor_creation_ops.py
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
test_tensorboard.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_tensorexpr_pybind.py
…
test_tensorexpr.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_testing.py
[1/N] Apply UP035 rule in tests (#163947)
2025-09-29 01:42:01 +00:00
test_throughput_benchmark.py
Fix Throughputbenchmark issue (#144669)
2025-01-26 03:37:20 +00:00
test_torch.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_torchfuzz_repros.py
Enable all SIM rules except disabled ones (#164645)
2025-10-17 07:27:11 +00:00
test_transformers.py
[cuDNN][SDPA] Handle noncontig nested tensors in cuDNN SDPA (#164958)
2025-10-09 21:58:54 +00:00
test_type_hints.py
Pyrefly suppressions 7/n (#164913)
2025-10-08 07:27:17 +00:00
test_type_info.py
Pyrefly suppressions 7/n (#164913)
2025-10-08 07:27:17 +00:00
test_type_promotion.py
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
test_typing.py
[BE][2/6] fix typos in test/ (test/test_*.py) (#157636)
2025-07-09 11:02:23 +00:00
test_unary_ufuncs.py
[BUG] Fix nonzero_static crash on CUDA when the input is a empty tensor (#162578)
2025-09-15 05:44:15 +00:00
test_utils_config_module.py
Add check that envvar configs are boolean (#145454)
2025-02-05 19:40:10 +00:00
test_utils_filelock.py
filelock: Make waitcounter variant to use (#139816)
2024-12-12 01:18:34 +00:00
test_utils.py
Revert "Add device argument to torch.random.get_rng_state (#163034)"
2025-10-04 15:25:45 +00:00
test_varlen_attention.py
varlen api (#164502)
2025-10-15 19:45:55 +00:00
test_view_ops.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_vulkan.py
Fix unused Python variables in test/[e-z]* (#136964)
2024-12-18 23:02:30 +00:00
test_weak.py
[BE][2/6] fix typos in test/ (test/test_*.py) (#157636)
2025-07-09 11:02:23 +00:00
test_xnnpack_integration.py
Enable all PIE rules on ruff (#165814)
2025-10-18 07:36:18 +00:00
test_xpu.py
Add a new API torch.xpu.is_tf32_supported for Intel GPU (#163141)
2025-10-12 12:11:57 +00:00