pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Manuel Candales aea57b3aa3 AOTI MPS Shim Implementation (#163865 )

## MPS Shim API

*   Updated MPS shimification API with handles and function declarations:
    *   `AOTIMetalShaderLibraryHandle` and `AOTIMetalKernelFunctionHandle` types
    *   Library management: `aoti_torch_mps_create_shader_library`, `aoti_torch_mps_delete_shader_library`, `aoti_torch_mps_get_kernel_function`
    *   Kernel execution: `aoti_torch_mps_run_command_block`, `aoti_torch_mps_start_encoding`, `aoti_torch_mps_dispatch` variants, etc

## MPS Shader Codegen

*   Modified to generate source constants instead of direct `DynamicMetalShaderLibrary` instantiation:
    *   **Before**: `at::native::mps::DynamicMetalShaderLibrary mps_lib_0(R"MTL(...)MTL");`
    *   **After**: `const char* mps_lib_0_source = R"MTL(...)MTL";`
*   Updated kernel call generation  to use shimified functions:
    *   Generates calls to shimified API instead of direct libtorch calls

## Before vs After Comparison

### Section 1: Shader Library
**Before (Direct Library Object)**
```cpp
at::native::mps::DynamicMetalShaderLibrary mps_lib_0(R"MTL(
    ...
)MTL");
```
**After (Source String)**
```cpp
const char* mps_lib_0_source = (R"MTL(
    ...
)MTL");
```

### Section 2: Getter Functions & RAII Management

**Before (Direct Library Access)**
```cpp
const std::shared_ptr<at::native::mps::MetalKernelFunction> get_mps_lib_0() {
    static const auto func = mps_lib_0.getKernelFunction("generated_kernel");
    return func;
}

AOTIMetalKernelFunctionHandle get_mps_lib_0_handle() {
    static const auto handle = AOTIMetalKernelFunctionHandle(get_mps_lib_0().get());
    return handle;
}
```

**After (Shim API + RAII Wrapper)**
```cpp
AOTIMetalKernelFunctionHandle get_mps_lib_0_handle() {
    static auto kernel_handle = []() {
        AOTIMetalShaderLibraryHandle lib_handle = nullptr;
        AOTIMetalKernelFunctionHandle kern_handle = nullptr;

        aoti_torch_mps_create_shader_library(mps_lib_0_source, &lib_handle);
        aoti_torch_mps_get_kernel_function(lib_handle, "generated_kernel", &kern_handle);

        // RAII wrapper with custom deleter
        auto lib_deleter = [](AOTIMetalShaderLibraryHandle h) {{
            if (h) aoti_torch_mps_delete_shader_library(h);
        }};

        using LibDeleter = decltype(lib_deleter);
        using LibPtr = std::unique_ptr<AOTIMetalShaderLibraryOpaque, LibDeleter>;

        // Return pair of kernel handle and library smart pointer for cleanup
        return std::make_pair(kern_handle, LibPtr(lib_handle, lib_deleter));
    }();
    return kernel_handle.first;
}
```

### Section 3: Runtime Execution

**Before (Direct Library Methods)**
```cpp
void AOTInductorModel::run_impl(...) {

    ...

    get_mps_lib_0()->runCommandBlock([&] {
        get_mps_lib_0()->startEncoding();
        aoti_torch_mps_set_arg_tensor(get_mps_lib_0_handle(), 0, buf0);
        aoti_torch_mps_set_arg_tensor(get_mps_lib_0_handle(), 1, arg0_1);
        aoti_torch_mps_set_arg_tensor(get_mps_lib_0_handle(), 2, arg1_1);
        get_mps_lib_0()->dispatch({static_cast<uint64_t>(10LL)});

    });

    ...

} // AOTInductorModel::run_impl
```

**After (Shim API with Lambda Pattern)**
```cpp
void AOTInductorModel::run_impl(...) {

    ...

    auto mps_lib_0_lambda_0 = [&](AOTIMetalKernelFunctionHandle handle) {
        aoti_torch_mps_start_encoding(handle);
        aoti_torch_mps_set_arg_tensor(handle, 0, buf0);
        aoti_torch_mps_set_arg_tensor(handle, 1, arg0_1);
        aoti_torch_mps_set_arg_tensor(handle, 2, arg1_1);
        aoti_torch_mps_dispatch_single(handle, static_cast<uint64_t>(10LL));
    };

    std::function<void(AOTIMetalKernelFunctionHandle)> mps_lib_0_func_wrapper_0 = mps_lib_0_lambda_0;
    aoti_torch_mps_run_command_block(get_mps_lib_0_handle(), aoti_torch_mps_shared_callback, &mps_lib_0_func_wrapper_0);

    ...

} // AOTInductorModel::run_impl
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163865
Approved by: https://github.com/angelayi, https://github.com/desertfire

2025-10-09 16:06:36 +00:00

ao/sparsity

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +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

Revert "[BC-Breaking] Remove long-deprecated casting functions from native_functions.yaml (#164641 )"

2025-10-09 15:42:51 +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

[OpenReg] Add AMP Integration guide for accelerators (#162050 )

2025-09-30 12:27:11 +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

Fix replacement reconstruct (#164937 )

2025-10-09 15:31:23 +00:00

distributions

[Reland][2/N]Port several test files under test/distributed to Intel GPU (#159473 )

2025-09-17 06:42:27 +00:00

dynamo

Introduce joint_custom_pass callback (#164981 )

2025-10-09 04:40:54 +00:00

dynamo_expected_failures

Add a RECORD_FUNCTION for Python fallback so it shows in profile (#160573 )

2025-10-01 14:10:44 +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_mm_v2 API (#164141 )

2025-10-09 12:43:18 +00:00

export

[export] Turn on install_free_tensors flag (#164691 )

2025-10-09 03:25:15 +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

Revert "Do not decompose in functionalization/proxy tensor if autograd wouldn't have decomposed (#164939 )"

2025-10-09 09:50:59 +00:00

use check_size instead of check_is_size in ops.py (#164668 )

2025-10-08 14:23:38 +00:00

higher_order_ops

use check_size instead of check_is_size in ops.py (#164668 )

2025-10-08 14:23:38 +00:00

inductor

AOTI MPS Shim Implementation (#163865 )

2025-10-09 16:06:36 +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

[reland] Allow setting grad_dtype on leaf tensors (#164751 )

2025-10-08 20:23:13 +00:00

jit_hooks

[Cmake] Make PyTorch buildable by CMake-4.x (#150203 )

2025-03-29 01:39:13 +00:00

lazy

[BE][PYFMT] migrate PYFMT for test/[i-z]*/ to ruff format (#144556 )

2025-07-29 03:26:09 +00:00

mobile

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +00:00

[ROCm][CI] test_convolution.py uses miopen immediate mode (#164598 )

2025-10-06 17:48:50 +00:00

onnx

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +00:00

optim

[optim] prevent unintended aliasing in lr_scheduler; update type annotations/docs (#163120 )

2025-09-25 06:58:58 +00:00

package

[Torch Package] Make get names of OrderedImporters support fallback to importers (#155743 )

2025-08-06 02:26:10 +00:00

profiler

More ruff SIM fixes (#164695 )

2025-10-09 03:24:50 +00:00

quantization

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +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

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +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

[Docs] Update FileCheck doc (#135199 )

2024-09-06 08:18:38 +00:00

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

[Code Clean] Remove support of python3.9 (#163846 )

2025-10-09 11:54:10 +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 (#164726 )

2025-10-06 12:57:29 +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

Enable TemporaryFileName tests on Windows (#146311 )

2025-02-07 06:06:18 +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

[reland] Allow setting grad_dtype on leaf tensors (#164751 )

2025-10-08 20:23:13 +00:00

test_autoload.py

…

test_binary_ufuncs.py

[TEST][CUDA] Use proper dtype in test_cuda_tensor_pow_scalar_tensor_cuda (#163070 )

2025-09-16 18:28:50 +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

Add split-K control to cuBLAS reduced-precision settings (#164766 )

2025-10-08 18:48:45 +00:00

test_custom_ops.py

[opaque obj] Error for torch.library.custom_op infer_schema (#163277 )

2025-10-08 09:09:16 +00:00

test_dataloader.py

Skip test_ind_worker_queue on Windows and macOS (flaky) (#162555 )

2025-09-10 07:05:14 +00:00

test_datapipe.py

[BC breaking] Remove deprecated imports for torch.utils.data.datapipes.iter.grouping (#163438 )

2025-09-23 05:02:06 +00:00

test_decomp.py

Revert "Do not decompose in functionalization/proxy tensor if autograd wouldn't have decomposed (#164939 )"

2025-10-09 09:50:59 +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

Upgrades dlpack to v1.1 to include fp8/fp4 (#162195 )

2025-09-17 16:39:11 +00:00

test_dynamic_shapes.py

Fix Avoid DDE in item numel check (#164934 )

2025-10-09 13:09:06 +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] Transformer/SDPA unit test parity (#163745 )

2025-09-25 17:14:19 +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

[fx] fix split_module with symint (#160093 )

2025-08-13 05:50:15 +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

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +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

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +00:00

test_itt.py

…

test_jit_autocast.py

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +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

Stop parsing command line arguments every time common_utils is imported. (#156703 )

2025-10-02 22:22:04 +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

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +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][CI] Add support for gfx1100 in rocm workflow + test skips (#148355 )

2025-10-07 22:36:25 +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

[ROCm][CI] Add support for gfx1100 in rocm workflow + test skips (#148355 )

2025-10-07 22:36:25 +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

[MPS] Add backward pass for embedding_bag (#163931 )

2025-10-03 00:48:38 +00:00

test_multiprocessing_spawn.py

Test multiprocessing spawn timing fix (#160672 )

2025-08-15 00:11:55 +00:00

test_multiprocessing.py

Revert "[CI] Use setup-python from for Mac tests (#155698 )"

2025-06-12 14:42:32 +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

Fix NestedTensor max/min operations for integer dtypes. (#162273 )

2025-10-02 18:46:27 +00:00

test_nn.py

[cuDNN][RNN] cuDNN RNN supports BFloat16 inputs since 9.13 (#164411 )

2025-10-08 15:26:50 +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

Revert "[RELAND] Always build USE_DISTRIBUTED (#160449 ) and Make distributed modules importable even when backend not built (#159889 ) (#162594 )"

2025-09-25 13:47:46 +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

Numpy zerotensor handling (#164487 )

2025-10-07 00:34:14 +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

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +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

Revert "Fix double dispatch to Python for detach (#163671 )"

2025-10-08 01:46:45 +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

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +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

Add scaled_mm python API, test (#164142 )

2025-10-09 12:43:18 +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

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +00:00

test_serialization.py

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +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

Remove old ROCm version check in tests (#164245 )

2025-10-06 22:42:01 +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 norm (#164961 )

2025-10-08 21:41: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

[4/N] Apply py39 ruff and pyupgrade fixes (#143257 )

2025-01-04 10:47:51 +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

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +00:00

test_tensorboard.py

Fix protobuf test comparison by parsing proto instead of raw strings (#162644 )

2025-09-12 16:26:54 +00:00

test_tensorexpr_pybind.py

…

test_tensorexpr.py

[BE][2/6] fix typos in test/ (test/test_*.py) (#157636 )

2025-07-09 11:02:23 +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

[cuBLAS] update cuBLAS determinism docs, remove workspace requirement checks (#161749 )

2025-10-03 00:09:47 +00:00

test_torchfuzz_repros.py

[torchfuzz] check in some more xfail repros (#164619 )

2025-10-06 16:20:44 +00:00

test_transformers.py

[cuDNN][SDPA] Disable dropout for cuDNN SDPA on 9.11 - 9.13 (#163903 )

2025-09-26 23:50:09 +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

[BE] Raise ValueError from torch.cat meta func (#158249 )

2025-07-20 23:49:18 +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_view_ops.py

unify broadcast_shapes functions and avoid duplicates (#160251 )

2025-08-16 00:54:32 +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

[BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408 )

2025-02-04 19:07:04 +00:00

test_xpu.py

Add a new API torch.xpu.can_device_access_peer for Intel GPU (#162705 )

2025-09-16 18:00:22 +00:00