| .. |
|
_nvfuser
|
|
|
|
ao/sparsity
|
|
|
|
autograd
|
|
|
|
backends/xeon
|
|
|
|
benchmark_utils
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
Switch dynamic_shapes to True by default (#103597)
|
2023-06-15 15:16:20 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
extend serialization for tensor metadata (#99808)
|
2023-06-14 01:43:21 +00:00 |
|
custom_backend
|
|
|
|
custom_operator
|
|
|
|
distributed
|
Inductor support for all_gather_into_tensor_coalesced. (#98643)
|
2023-06-21 19:25:03 +00:00 |
|
distributions
|
Fix Dirichlet.log_prob() when x=0 and alpha=1 (#103605)
|
2023-06-15 16:16:50 +00:00 |
|
dynamo
|
Reland of https://github.com/pytorch/pytorch/pull/101818 (#103888)
|
2023-06-21 21:00:56 +00:00 |
|
edge
|
[Reland][ET] Select used et_kernel_metadata only (#103705)
|
2023-06-18 00:33:28 +00:00 |
|
error_messages
|
|
|
|
expect
|
[decomp] Add decomposition for torch.renorm (#103858)
|
2023-06-21 20:57:43 +00:00 |
|
export
|
Lift user defined attributes into inputs for certain cases (user defined types and tensors) (#103386)
|
2023-06-20 23:45:19 +00:00 |
|
forward_backward_compatibility
|
Implement adding bias vector into structured sparse linear operator (#100881)
|
2023-06-15 16:16:09 +00:00 |
|
functorch
|
[decomp] Add decomposition for torch.renorm (#103858)
|
2023-06-21 20:57:43 +00:00 |
|
fx
|
|
|
|
inductor
|
[inductor] Lower diagonal, diagonal_copy and diagonal_scatter (#103755)
|
2023-06-21 20:16:24 +00:00 |
|
jit
|
Fix shape function for transpose convolution (#102139)
|
2023-06-21 17:50:56 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
|
|
|
mobile
|
|
|
|
nn
|
Lift user defined attributes into inputs for certain cases (user defined types and tensors) (#103386)
|
2023-06-20 23:45:19 +00:00 |
|
onnx
|
[ONNX] FX Dispatcher Test (#103971)
|
2023-06-21 19:04:08 +00:00 |
|
onnx_caffe2
|
|
|
|
optim
|
Fix optim state_dict casting to allow step to cast to CPU (#102619)
|
2023-06-13 00:46:40 +00:00 |
|
package
|
|
|
|
profiler
|
[Profiler] Fix flaky test_memory_timeline_no_id (#103441)
|
2023-06-12 22:58:56 +00:00 |
|
quantization
|
[codemod][numpy] replace np.str with str (#103931)
|
2023-06-21 18:16:42 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
typing
|
|
|
|
_test_bazel.py
|
|
|
|
allowlist_for_publicAPI.json
|
|
|
|
conftest.py
|
Do not collect and skip non-disabled tests when rerunning disabled tests (#102107)
|
2023-05-27 12:10:36 +00:00 |
|
create_dummy_torchscript_model.py
|
|
|
|
delete.py
|
|
|
|
HowToWriteTestsUsingFileCheck.md
|
|
|
|
linear.py
|
|
|
|
load_torchscript_model.py
|
|
|
|
mkl_verbose.py
|
|
|
|
mkldnn_verbose.py
|
|
|
|
run_doctests.sh
|
|
|
|
run_test.py
|
Enable uploading metrics and upload Test Reordering metrics to dynamodb (#102691)
|
2023-06-12 23:01:53 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
|
|
|
test_autocast.py
|
[BE] Deprecate has_XYZ attributes (#103279)
|
2023-06-10 05:17:17 +00:00 |
|
test_autograd.py
|
Revert "Record view stacks if running anomaly mode (#103185)"
|
2023-06-12 23:52:10 +00:00 |
|
test_binary_ufuncs.py
|
|
|
|
test_bundled_images.py
|
|
|
|
test_bundled_inputs.py
|
|
|
|
test_comparison_utils.py
|
|
|
|
test_compile_benchmark_util.py
|
|
|
|
test_complex.py
|
|
|
|
test_content_store.py
|
|
|
|
test_cpp_api_parity.py
|
|
|
|
test_cpp_extensions_aot.py
|
[MPS] Prerequisite for MPS C++ extension (#102483)
|
2023-06-07 17:28:31 +00:00 |
|
test_cpp_extensions_jit.py
|
[MPS] Prerequisite for MPS C++ extension (#102483)
|
2023-06-07 17:28:31 +00:00 |
|
test_cpp_extensions_open_device_registration.py
|
extend serialization for tensor metadata (#99808)
|
2023-06-14 01:43:21 +00:00 |
|
test_cuda.py
|
[memory_viz] Unified viewer (#103565)
|
2023-06-16 03:49:48 +00:00 |
|
test_cuda_expandable_segments.py
|
|
|
|
test_cuda_nvml_based_avail.py
|
|
|
|
test_cuda_primary_ctx.py
|
|
|
|
test_cuda_sanitizer.py
|
|
|
|
test_cuda_trace.py
|
|
|
|
test_custom_op_testing.py
|
[operator_compile_check] Add FakeTensor testing (#103595)
|
2023-06-16 16:55:51 +00:00 |
|
test_dataloader.py
|
|
|
|
test_datapipe.py
|
|
|
|
test_decomp.py
|
|
|
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
|
|
|
test_dlpack.py
|
add additional stream priority for cuda streams (#101956)
|
2023-05-27 02:36:16 +00:00 |
|
test_dynamic_shapes.py
|
fix soundness bug with unsupported constraints (#102897)
|
2023-06-10 01:59:55 +00:00 |
|
test_expanded_weights.py
|
tf32 context fixes for various tests (#103137)
|
2023-06-15 02:33:12 +00:00 |
|
test_fake_tensor.py
|
[operator_compile_check] Add FakeTensor testing (#103595)
|
2023-06-16 16:55:51 +00:00 |
|
test_flop_counter.py
|
De-register forward hooks upon exiting flop counter context (#103744)
|
2023-06-20 19:34:02 +00:00 |
|
test_foreach.py
|
Reland "Move tensor grouping to ATen" (#103912)
|
2023-06-21 09:26:33 +00:00 |
|
test_function_schema.py
|
|
|
|
test_functional_autograd_benchmark.py
|
|
|
|
test_functional_optim.py
|
|
|
|
test_functionalization.py
|
|
|
|
test_functionalization_of_rng_ops.py
|
|
|
|
test_futures.py
|
|
|
|
test_fx.py
|
Preserve original co_filename when FX symbolic_trace (#103885)
|
2023-06-21 08:28:50 +00:00 |
|
test_fx_experimental.py
|
|
|
|
test_fx_passes.py
|
|
|
|
test_fx_reinplace_pass.py
|
|
|
|
test_hub.py
|
|
|
|
test_import_stats.py
|
|
|
|
test_indexing.py
|
[Dynamo] Fix lineinfo generation on PY3.11+ (#103525)
|
2023-06-14 05:41:43 +00:00 |
|
test_itt.py
|
|
|
|
test_jit.py
|
|
|
|
test_jit_autocast.py
|
|
|
|
test_jit_cuda_fuser.py
|
|
|
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
|
|
|
test_jit_legacy.py
|
|
|
|
test_jit_llga_fuser.py
|
[BE] Deprecate has_XYZ attributes (#103279)
|
2023-06-10 05:17:17 +00:00 |
|
test_jit_profiling.py
|
|
|
|
test_jit_simple.py
|
|
|
|
test_jit_string.py
|
|
|
|
test_jiterator.py
|
|
|
|
test_kernel_launch_checks.py
|
|
|
|
test_legacy_vmap.py
|
Make index_add_ error if input source shape is wrong (#100321)
|
2023-06-08 06:51:10 +00:00 |
|
test_license.py
|
|
|
|
test_linalg.py
|
Enable addmm + GELU epilogue fusion via cuBLASLt (#103811)
|
2023-06-21 19:59:40 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
|
|
|
test_maskedtensor.py
|
|
|
|
test_matmul_cuda.py
|
|
|
|
test_meta.py
|
[pt2] add meta and SymInt support for linalg_matrix_exp (#102945)
|
2023-06-09 22:45:16 +00:00 |
|
test_metal.py
|
|
|
|
test_mkl_verbose.py
|
|
|
|
test_mkldnn.py
|
[BE] Deprecate has_XYZ attributes (#103279)
|
2023-06-10 05:17:17 +00:00 |
|
test_mkldnn_fusion.py
|
[BE] Deprecate has_XYZ attributes (#103279)
|
2023-06-10 05:17:17 +00:00 |
|
test_mkldnn_verbose.py
|
|
|
|
test_mobile_optimizer.py
|
|
|
|
test_model_dump.py
|
|
|
|
test_module_init.py
|
|
|
|
test_modules.py
|
Remove a number of fixed skips (#103162)
|
2023-06-08 17:37:59 +00:00 |
|
test_monitor.py
|
|
|
|
test_mps.py
|
[MPS] Handle deserialization more permissively (#98834)
|
2023-06-15 15:51:03 +00:00 |
|
test_multiprocessing.py
|
|
|
|
test_multiprocessing_spawn.py
|
|
|
|
test_namedtensor.py
|
|
|
|
test_namedtuple_return_api.py
|
|
|
|
test_native_functions.py
|
|
|
|
test_native_mha.py
|
|
|
|
test_nestedtensor.py
|
|
|
|
test_nn.py
|
Add assign kwarg to module.load_state_dict (#102212)
|
2023-06-15 18:41:00 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
|
|
|
test_nvfuser_dynamo.py
|
|
|
|
test_nvfuser_frontend.py
|
|
|
|
test_openmp.py
|
|
|
|
test_ops.py
|
Tidy __all__ under torch._refs (#103712)
|
2023-06-20 00:04:58 +00:00 |
|
test_ops_fwd_gradients.py
|
|
|
|
test_ops_gradients.py
|
|
|
|
test_ops_jit.py
|
|
|
|
test_optim.py
|
|
|
|
test_overrides.py
|
|
|
|
test_package.py
|
|
|
|
test_per_overload_api.py
|
|
|
|
test_prims.py
|
Replace _prims_common.check with torch._check* (#103240)
|
2023-06-21 00:46:17 +00:00 |
|
test_proxy_tensor.py
|
Reland of https://github.com/pytorch/pytorch/pull/101818 (#103888)
|
2023-06-21 21:00:56 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
|
|
|
test_python_dispatch.py
|
Add API to construct the functional variant of an op (#102293)
|
2023-06-02 13:36:50 +00:00 |
|
test_pytree.py
|
improve repr for pytrees (#103945)
|
2023-06-21 20:53:03 +00:00 |
|
test_quantization.py
|
[Quant][PT2E]Enable x86 inductor quantizer (#98730)
|
2023-06-17 06:10:23 +00:00 |
|
test_reductions.py
|
|
|
|
test_scatter_gather_ops.py
|
|
|
|
test_schema_check.py
|
|
|
|
test_segment_reductions.py
|
|
|
|
test_serialization.py
|
Add mmap option to torch.load (#102549)
|
2023-06-09 15:49:58 +00:00 |
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
|
|
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
Make 1D integer sorting work in parallel (#100081)
|
2023-06-02 07:41:28 +00:00 |
|
test_sparse.py
|
Add activation functions (ReLU and SiLU for now) for structured sparse linear operator (#101339)
|
2023-06-16 17:24:59 +00:00 |
|
test_sparse_csr.py
|
softmax: Triton kernel for BSR inputs (#102095)
|
2023-06-21 01:23:27 +00:00 |
|
test_spectral_ops.py
|
|
|
|
test_stateless.py
|
|
|
|
test_static_runtime.py
|
|
|
|
test_subclass.py
|
|
|
|
test_sympy_utils.py
|
|
|
|
test_tensor_creation_ops.py
|
|
|
|
test_tensorboard.py
|
|
|
|
test_tensorexpr.py
|
|
|
|
test_tensorexpr_pybind.py
|
|
|
|
test_testing.py
|
|
|
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
Make torch.empty* deterministic by filling with NaN or max int value (#101849)
|
2023-06-21 02:53:22 +00:00 |
|
test_transformers.py
|
Back out "Dropout support for memory efficient attention (#102038)" & "Two small mem_eff bug fixes (#103201)" (#103464)
|
2023-06-12 18:56:48 -07:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
|
|
|
test_type_promotion.py
|
|
|
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
add Half support for logsigmoid, threshold, elu, gelu, hardtanh, hardsigmoid, hardswish, hardshrink, softshrink, leakyrelu, softplus, glu, silu, mish, and prelu on CPU (#98745)
|
2023-05-27 16:20:21 +00:00 |
|
test_utils.py
|
Add torch._utils.render_call, improve printoptions (#102623)
|
2023-05-31 22:08:04 +00:00 |
|
test_view_ops.py
|
|
|
|
test_vulkan.py
|
|
|
|
test_weak.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
