pytorch

ao/sparsity

[BE]: Update flake8 to v6.1.0 and fix lints (#116591 )

2024-01-03 06:04:44 +00:00

autograd

Autograd attaches logging hooks only in debug level (#116522 )

2024-01-02 20:06:18 +00:00

backends/xeon

enable taskset core pinning in addition to numactl (#96011 )

2023-03-07 01:19:46 +00:00

benchmark_utils

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

bottleneck_test

…

cpp

Check QNNPACK support for the platform before running test (#119139 )

2024-02-12 20:21:07 +00:00

cpp_api_parity

Add python and C++ support for LPPool3d (#114199 )

2023-12-08 18:18:44 +00:00

cpp_extensions

Enable optional tensorList fallback to cpu. (#119273 )

2024-02-07 03:54:13 +00:00

custom_backend

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

custom_operator

Better unsupported op error message (#117770 )

2024-01-23 15:05:16 +00:00

distributed

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

distributions

Bugfix to MixtureSameFamily's _pad_mixture_dimension (#118947 )

2024-02-06 16:24:22 +00:00

dynamo

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

edge

Adding event_tracer evalue logging calls in codegen (#114584 )

2023-11-28 18:32:05 +00:00

error_messages

[BE] Enable ruff's UP rules and autoformat dynamo / functorch and refs (#105432 )

2023-07-19 13:48:44 +00:00

expect

Add lowerings to special functions (#119187 )

2024-02-11 16:35:40 +00:00

export

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

forward_backward_compatibility

Revert "[RELAND] Remove deprecated fbgemm operators (#112153 )"

2024-02-01 18:35:19 +00:00

functorch

Fixed hash issue in fx_graph_cse (#119567 )

2024-02-12 18:52:11 +00:00

fx

LazyGraphModule (#117911 )

2024-01-27 04:10:18 +00:00

inductor

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

jit

[BE]: Apply RUF025 dict.fromkeys preview rule (#118637 )

2024-01-30 20:46:54 +00:00

jit_hooks

…

lazy

Implement shallow copy functions for FunctionalTensorWrapper. (#118783 )

2024-02-08 17:15:46 +00:00

mobile

Add some tests for 3d channels last (#118283 )

2024-01-30 01:26:47 +00:00

nn

Integrate swap_tensors into nn.Module.load_state_dict (#117913 )

2024-02-09 22:32:29 +00:00

onnx

[ONNX] Support op consistency error reproduction (#119512 )

2024-02-09 23:24:01 +00:00

onnx_caffe2

Revert "[ONNX] Remove the depreacated function _export (#109763 )"

2023-09-25 17:47:21 +00:00

optim

ReduceLROnPlateau init _last_lr (#119366 ) (#119556 )

2024-02-09 19:35:02 +00:00

package

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

profiler

Fix deadlock in ExecutionTraceObserver (#119242 ) (#119398 )

2024-02-08 18:00:51 +00:00

quantization

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

scripts

Fix flake8 lint errors reported by ruff - take 2 (#99798 )

2023-04-23 23:09:51 +00:00

test_img

…

torch_np

fix is_ and is_not (#118978 )

2024-02-12 19:04:40 +00:00

typing

Fix signatures of torch.{add, sub, mul} (#118398 )

2024-01-30 22:18:15 +00:00

_test_bazel.py

[bazel] add build for functorch (#101475 )

2023-05-18 20:29:08 +00:00

allowlist_for_publicAPI.json

Autograd doc cleanup (#118500 )

2024-01-29 21:51:33 +00:00

conftest.py

Various CI settings (#117668 )

2024-01-26 00:17:29 +00:00

create_dummy_torchscript_model.py

Add basic Module serialization BC test (#96238 )

2023-03-08 21:01:27 +00:00

delete.py

…

HowToWriteTestsUsingFileCheck.md

…

linear.py

…

load_torchscript_model.py

Add basic Module serialization BC test (#96238 )

2023-03-08 21:01:27 +00:00

minioptest_failures_dict.json

[generate_opcheck_tests] Enable using same failures_dict for multiple testclasses (#110164 )

2023-09-28 17:56:45 +00:00

mkl_verbose.py

…

mkldnn_verbose.py

…

pytest_shard_custom.py

Reduce pytest prints (#117069 )

2024-01-23 18:39:30 +00:00

run_doctests.sh

…

run_test.py

No TD (test removal) option in CI (#118808 )

2024-02-09 16:42:27 +00:00

simulate_nccl_errors.py

Fix G001,G002,G003 in logs to % syntax (#97812 )

2023-04-01 01:43:33 +00:00

test_ao_sparsity.py

add pruning method: Filter Pruning via Geometric Median for Deep Convolutional Neural Networks Acceleration (#95689 )

2023-08-02 16:24:42 +00:00

test_autocast.py

Remove incorrect usages of skipIfTorchDynamo (#117114 )

2024-01-10 22:25:31 +00:00

test_autograd_fallback.py

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

test_autograd.py

Revert "Reify view_func() closures as ViewFuncs (#118404 )"

2024-02-12 12:38:51 +00:00

test_binary_ufuncs.py

[Dynamo] Add complex to literal constant (#117819 )

2024-01-23 23:46:46 +00:00

test_bundled_images.py

[BE]: Update flake8 to v6.1.0 and fix lints (#116591 )

2024-01-03 06:04:44 +00:00

test_bundled_inputs.py

…

test_comparison_utils.py

Add missing __main__ in two unittests (#97302 )

2023-03-22 19:09:08 +00:00

test_compile_benchmark_util.py

unify TEST_CUDA (#106685 )

2023-08-10 09:01:36 +00:00

test_complex.py

Don't fastpath conj copy when conj/neg bit mismatch (#108881 )

2023-09-08 20:44:43 +00:00

test_content_store.py

Windows Dynamo Error Removal CI Check (#115969 )

2024-02-08 21:23:45 +00:00

test_cpp_api_parity.py

markDynamoStrictTest some test_cpp_extensions.* (#115858 )

2023-12-15 01:22:38 +00:00

test_cpp_extensions_aot.py

markDynamoStrictTest some test_cpp_extensions.* (#115858 )

2023-12-15 01:22:38 +00:00

test_cpp_extensions_jit.py

markDynamoStrictTest some test_cpp_extensions.* (#115858 )

2023-12-15 01:22:38 +00:00

test_cpp_extensions_open_device_registration.py

Enable optional tensorList fallback to cpu. (#119273 )

2024-02-07 03:54:13 +00:00

test_cuda_expandable_segments.py

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

test_cuda_multigpu.py

add GradScaler on CPU (#109993 )

2024-01-29 23:42:35 +00:00

test_cuda_nvml_based_avail.py

markDynamoStrictTest some more tests (#115857 )

2023-12-15 01:22:38 +00:00

test_cuda_primary_ctx.py

markDynamoStrictTest some more tests (#115857 )

2023-12-15 01:22:38 +00:00

test_cuda_sanitizer.py

unify TEST_CUDA (#106685 )

2023-08-10 09:01:36 +00:00

test_cuda_trace.py

markDynamoStrictTest some more tests (#115857 )

2023-12-15 01:22:38 +00:00

test_cuda.py

add test cases for GradScaler on CPU (#109994 )

2024-02-02 21:49:07 +00:00

test_custom_ops.py

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

test_dataloader.py

[BE]: Apply RUF025 dict.fromkeys preview rule (#118637 )

2024-01-30 20:46:54 +00:00

test_datapipe.py

Replace follow_imports = silent with normal (#118414 )

2024-01-27 02:44:11 +00:00

test_decomp.py

[decomp] Add tests for different dtypes to SDPA decomposition (#119239 )

2024-02-06 11:17:07 +00:00

test_deploy.py

…

test_determination.py

…

test_dispatch.py

Revert "markDynamoStrictTest more tests (#115871 )"

2023-12-19 15:36:27 +00:00

test_dlpack.py

Improve uint{16,32,64} dlpack/numpy compatibility (#116808 )

2024-01-11 17:01:54 +00:00

test_dynamic_shapes.py

Avoid performing replacements when it would unrefine ranges (#117356 )

2024-02-09 14:43:58 +00:00

test_expanded_weights.py

Make variables in dict LazyTrackers (not lazily guarded yet) and avoid using DICT_KEYS guard (#117625 )

2024-02-02 14:38:08 +00:00

test_fake_tensor.py

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

test_flop_counter.py

Re-add initial Flash Attention support on ROCM (#115981 )

2024-01-04 22:21:31 +00:00

test_foreach.py

Refactor some tests by using TEST_CUDA & TEST_MULTIGPU instead (#116083 )

2024-01-03 08:53:59 +00:00

test_function_schema.py

…

test_functional_autograd_benchmark.py

…

test_functional_optim.py

Cudagraphs support for compiled optimizers (#107504 )

2023-08-31 20:47:18 +00:00

test_functionalization_of_rng_ops.py

[philox_rand] Add decomps (#100206 )

2023-04-28 02:20:13 +00:00

test_functionalization.py

Enable reverse view_funcs by default for python subclasses (#116512 )

2024-01-05 16:48:12 +00:00

test_futures.py

[Reland] Update mypy to 1.4.1 (#105227 )

2023-07-15 20:30:20 +00:00

test_fx_experimental.py

Revert "[codemod] markDynamoStrictTest batch 16 (#117218 )"

2024-01-12 03:06:20 +00:00

test_fx_passes.py

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

test_fx_reinplace_pass.py

…

test_fx.py

[codemod] markDynamoStrictTest batch 22 (#117729 )

2024-01-18 16:59:26 +00:00

test_hub.py

Default permissions for torch.hub downloads (#82869 )

2023-08-24 15:48:24 +00:00

test_import_stats.py

…

test_indexing.py

Support builtin callable with object arguments in dynamo (#118678 )

2024-01-31 17:54:08 +00:00

test_itt.py

…

test_jit_autocast.py

[BE] Enable C419 rule for any all shortcircuiting (#99890 )

2023-04-25 15:02:13 +00:00

test_jit_disabled.py

…

test_jit_fuser_legacy.py

…

test_jit_fuser_te.py

Remove unnecessary skipIfTorchDynamo from test_jit_fuser_te (#118728 )

2024-02-12 20:55:29 +00:00

test_jit_fuser.py

[BE]: Enable F821 and fix bugs (#116579 )

2024-01-01 08:40:46 +00:00

test_jit_legacy.py

…

test_jit_llga_fuser.py

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

test_jit_profiling.py

…

test_jit_simple.py

…

test_jit_string.py

…

test_jit.py

Remove unnecessary skipIfTorchDynamo from test_jit_fuser_te (#118728 )

2024-02-12 20:55:29 +00:00

test_jiterator.py

[ROCm] add skipCUDAIfVersionLessThan to unskip test_jiterator for ROCm (#99197 )

2023-04-17 16:05:16 +00:00

test_kernel_launch_checks.py

…

test_legacy_vmap.py

Skip some slow tests (under Dynamo) (#117389 )

2024-01-12 22:18:07 +00:00

test_license.py

…

test_linalg.py

[ROCm] enable hipsolver backend for linalg.eigh (#115177 )

2024-02-08 22:03:27 +00:00

test_logging.py

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

test_masked.py

…

test_maskedtensor.py

Tighten FakeTensor reentrancy asserts, add debugging (#102091 )

2023-05-24 05:37:51 +00:00

test_matmul_cuda.py

Enable scaled_mm on sm89 devices (#118881 )

2024-02-03 00:44:03 +00:00

test_meta.py

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

test_metal.py

…

test_mkl_verbose.py

…

test_mkldnn_fusion.py

[codemod] markDynamoStrictTest batch 22 (#117729 )

2024-01-18 16:59:26 +00:00

test_mkldnn_verbose.py

…

test_mkldnn.py

[BE]: Enable RUFF PERF402 and apply fixes (#115505 )

2023-12-20 18:01:24 +00:00

test_mobile_optimizer.py

Revert "markDynamoStrictTest on more tests (#115879 )"

2023-12-15 20:00:09 +00:00

test_model_dump.py

…

test_model_exports_to_core_aten.py

Build and test ExecuTorch on PyTorch (#113364 )

2023-11-16 02:19:58 +00:00

test_modules.py

[easy] Add testing utilties for torch.nn.utils.set_swap_module_params_on_conversion (#118023 )

2024-02-07 18:55:44 +00:00

test_monitor.py

Change dynamo_test_failures.py to silently run skipped tests (#117401 )

2024-01-17 02:48:19 +00:00

test_mps.py

[MPS][EZ] Fix cfloat->chalf conversion on MacOS13 (#119681 )

2024-02-12 19:09:10 +00:00

test_multiprocessing_spawn.py

Revert "markDynamoStrictTest on more tests (#115879 )"

2023-12-15 20:00:09 +00:00

test_multiprocessing.py

Revert "markDynamoStrictTest on more tests (#115879 )"

2023-12-15 20:00:09 +00:00

test_namedtensor.py

Skip test_none_names_refcount under Dynamo-wrapped CI (#118309 )

2024-01-25 22:21:22 +00:00

test_namedtuple_return_api.py

Revert "[CUDNN][SDPA] Experimental cuDNN Flash Attention v2 Inference (#115663 )"

2024-01-18 23:40:30 +00:00

test_native_functions.py

[Dynamo] Fix nested function resume execution (#100426 )

2023-05-11 03:10:23 +00:00

test_native_mha.py

Re-add initial Flash Attention support on ROCM (#115981 )

2024-01-04 22:21:31 +00:00

test_nestedtensor.py

Revert "Fix jagged NT softmax semantics (#119459 )"

2024-02-09 02:31:49 +00:00

test_nn.py

Fixed an issue where nn.Linear would cause an internal int underflow … (#119221 )

2024-02-08 21:06:34 +00:00

test_nnapi.py

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

test_numba_integration.py

…

test_numpy_interop.py

Replace follow_imports = silent with normal (#118414 )

2024-01-27 02:44:11 +00:00

test_openmp.py

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

test_ops_fwd_gradients.py

unMarkDynamoStrictTest on OpInfo-based tests (#115856 )

2023-12-15 01:22:31 +00:00

test_ops_gradients.py

unMarkDynamoStrictTest on OpInfo-based tests (#115856 )

2023-12-15 01:22:31 +00:00

test_ops_jit.py

unMarkDynamoStrictTest on OpInfo-based tests (#115856 )

2023-12-15 01:22:31 +00:00

test_ops.py

Revert "Reify view_func() closures as ViewFuncs (#118404 )"

2024-02-12 12:38:51 +00:00

test_optim.py

Migrate load_state_dict hook tests to OptimizerInfo (#119310 )

2024-02-07 16:00:01 +00:00

test_out_dtype_op.py

[export] Remove torch._export.export (#119095 )

2024-02-08 21:22:04 +00:00

test_overrides.py

[BE]: Update flake8 to v6.1.0 and fix lints (#116591 )

2024-01-03 06:04:44 +00:00

test_package.py

Reduce pytest blocklist part 2 (#96397 )

2023-03-10 19:10:43 +00:00

test_per_overload_api.py

…

test_prims.py

Revert "markDynamoStrictTest some more (#115885 )"

2023-12-15 19:51:24 +00:00

test_proxy_tensor.py

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

test_pruning_op.py

[dynamo] Eagerly install guards (#111415 )

2023-11-07 19:55:19 +00:00

test_public_bindings.py

Add torch.dtype instances to the public API (#119307 )

2024-02-07 02:57:49 +00:00

test_python_dispatch.py

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

test_pytree.py

[BE]: Apply RUF025 dict.fromkeys preview rule (#118637 )

2024-01-30 20:46:54 +00:00

test_quantization.py

enable cat for cuda bits types (#115044 )

2023-12-06 00:05:18 +00:00

test_reductions.py

Make and/or on uint8 tensors properly return 0x00 or 0x01 (#117827 )

2024-01-22 17:30:22 +00:00

test_scatter_gather_ops.py

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

test_schema_check.py

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

test_segment_reductions.py

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

test_serialization.py

[CI] Install dill in ci (#116214 )

2024-01-24 23:42:35 +00:00

test_set_default_mobile_cpu_allocator.py

…

test_shape_ops.py

…

test_show_pickle.py

Revert "markDynamoStrictTest some more (#115885 )"

2023-12-15 19:51:24 +00:00

test_sort_and_select.py

Removed an internal assertion for the optional stable value and inste… (#117414 )

2024-01-17 02:25:21 +00:00

test_sparse_csr.py

[SparseCsr] Remove triton sdpa skip after triton pin update (#109601 )

2024-02-08 16:40:25 +00:00

test_sparse_semi_structured.py

[sparse] Add in out_dtype support (i8i8->bf16, i32) for cusparselt (#119296 )

2024-02-12 16:02:36 +00:00

test_sparse.py

Add batched sparse CSR/CSC/BSR/BSC to sparse COO conversion support (#116206 )

2024-01-07 19:42:02 +00:00

test_spectral_ops.py

FFT: Handle noop fftn calls gracefully (#117368 )

2024-01-12 20:16:50 +00:00

test_stateless.py

Throw error if stateless.functional_call called with nn.DataParallel (#107403 )

2023-08-18 03:02:04 +00:00

test_static_runtime.py

Skip test_fork_wait_4 and test_fork_wait_4_async (#112743 )

2023-11-02 20:46:29 +00:00

test_subclass.py

Allow storage() to work on python tensor subclasses, but error on future data accesses (#107417 )

2023-08-22 15:25:31 +00:00

test_sympy_utils.py

Run sympy expressions with Python values / FX tracing (#113978 )

2023-11-20 21:25:11 +00:00

test_tensor_creation_ops.py

Unskip test_complex_type_conversions (#118694 )

2024-01-31 08:04:15 +00:00

test_tensorboard.py

Add tensorboard to pip requirements (#109349 )

2023-09-15 10:39:48 +00:00

test_tensorexpr_pybind.py

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

test_tensorexpr.py

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

test_testing.py

[BE]: Apply RUF025 dict.fromkeys preview rule (#118637 )

2024-01-30 20:46:54 +00:00

test_throughput_benchmark.py

…

test_torch.py

Update to TorchFix 0.4.0 (#119424 )

2024-02-12 23:30:12 +00:00

test_transformers.py

add Half support for flash attention (#119247 )

2024-02-07 05:57:41 +00:00

test_type_hints.py

…

test_type_info.py

Add finfo properties for float8 dtypes (#109744 )

2023-09-21 03:41:48 +00:00

test_type_promotion.py

Improve uint{16,32,64} dlpack/numpy compatibility (#116808 )

2024-01-11 17:01:54 +00:00

test_typing.py

More progress on type checking ValueRanges (#118870 )

2024-02-05 20:29:25 +00:00

test_unary_ufuncs.py

Remove some unnecessary skipIfTorchDynamo (#118725 )

2024-01-31 18:18:17 +00:00

test_utils.py

[ROCm] Hipify trie re-engineering and adding unit tests (#118433 )

2024-02-02 16:04:59 +00:00

test_view_ops.py

Implement numpy(force=True) (#109636 )

2023-09-20 20:06:13 +00:00

test_vulkan.py

…

test_weak.py

[BE]: Enable RUF015 codebase wide (#115507 )

2023-12-11 15:51:01 +00:00

test_xnnpack_integration.py

[BE] Enable ruff's UP rules and autoformat test/ (#105434 )

2023-07-19 20:36:06 +00:00

test_xpu.py

[2/2] Intel GPU Runtime Upstreaming for Stream (#117619 )

2024-02-10 03:39:42 +00:00