| .. |
|
_nvfuser
|
|
|
|
ao/sparsity
|
|
|
|
autograd
|
|
|
|
backends/xeon
|
|
|
|
benchmark_utils
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
Revert "replace torch::make_unique with std::make_unique (#108866)"
|
2023-09-17 21:57:30 +00:00 |
|
cpp_api_parity
|
[BE]: Update ruff to 0.285 (#107519)
|
2023-08-22 23:16:38 +00:00 |
|
cpp_extensions
|
|
|
|
custom_backend
|
|
|
|
custom_operator
|
|
|
|
distributed
|
[2D][1/N] Add _enable_extension to fsdp state (#109242)
|
2023-09-16 19:03:10 +00:00 |
|
distributions
|
[BE]: enable ruff rules PLR1722 and PLW3301 (#109461)
|
2023-09-18 02:07:21 +00:00 |
|
dynamo
|
[RELAND] Force synced KJT to trace unbacked SymInt (#108960) (#109216)
|
2023-09-18 14:39:44 +00:00 |
|
edge
|
|
|
|
error_messages
|
|
|
|
expect
|
[Decomposition] clamp_min (#108717)
|
2023-09-18 12:43:58 +00:00 |
|
export
|
[export] Preserve shape dynamism for unused inputs. (#109239)
|
2023-09-14 07:43:36 +00:00 |
|
forward_backward_compatibility
|
Flash Attention v2 (#105602)
|
2023-09-13 13:59:05 +00:00 |
|
functorch
|
Add meta function for _segment_reduce (#109359)
|
2023-09-16 13:31:03 +00:00 |
|
fx
|
Back out "Revert D49107540: [pytorch][PR] split by tag" (#109332)
|
2023-09-16 05:29:16 +00:00 |
|
inductor
|
Fix spelling / capitalization in freezing.py error message (#109347)
|
2023-09-18 18:12:20 +00:00 |
|
jit
|
[jit] Skip NNAPI, test_ivalue, CPU NNC tests in fbcode (#108937)
|
2023-09-11 22:42:30 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
|
|
|
mobile
|
|
|
|
nn
|
LazyBatchNorm{1-3}d support dict&set (#109015)
|
2023-09-12 09:09:59 +00:00 |
|
onnx
|
[ONNX] Fix indexing issue of meshgrid op (#109350)
|
2023-09-15 19:49:43 +00:00 |
|
onnx_caffe2
|
[ONNX] Remove deprecated functions (#107208)
|
2023-09-14 19:09:56 +00:00 |
|
optim
|
Add capturable ASGD impl (#107857)
|
2023-09-07 06:30:30 +00:00 |
|
package
|
|
|
|
profiler
|
Increased logging threshold for profiler matching (#108010)
|
2023-08-30 17:16:36 +00:00 |
|
quantization
|
[quant][be] Move QAT tests to its own file (#108061)
|
2023-09-15 18:34:44 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
torch_np
|
Dispatch numpy.take_along_axis to torch.take_along_dim (#108880)
|
2023-09-13 23:13:09 +00:00 |
|
typing
|
Improve type annotations for jit.script (#108782)
|
2023-09-13 19:20:25 +00:00 |
|
_test_bazel.py
|
|
|
|
allowlist_for_publicAPI.json
|
|
|
|
conftest.py
|
|
|
|
create_dummy_torchscript_model.py
|
|
|
|
delete.py
|
|
|
|
HowToWriteTestsUsingFileCheck.md
|
|
|
|
linear.py
|
|
|
|
load_torchscript_model.py
|
|
|
|
minioptest_failures_dict.json
|
[optests] Changed failures_dict format to json; automatic update of failures_dict (#109110)
|
2023-09-13 13:24:15 +00:00 |
|
mkl_verbose.py
|
|
|
|
mkldnn_verbose.py
|
|
|
|
run_doctests.sh
|
|
|
|
run_test.py
|
inductor/test_max_autotune serial in CI (#109209)
|
2023-09-13 17:04:43 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
|
|
|
test_autocast.py
|
Allow float dtype when Autocast CPU Disabled (#107348)
|
2023-09-01 00:49:44 +00:00 |
|
test_autograd.py
|
Update CopySlices to not internal assert when grad_output is undefined (#108353)
|
2023-09-11 16:26:05 +00:00 |
|
test_autograd_fallback.py
|
|
|
|
test_binary_ufuncs.py
|
[BE]: Update ruff to 0.285 (#107519)
|
2023-08-22 23:16:38 +00:00 |
|
test_bundled_images.py
|
|
|
|
test_bundled_inputs.py
|
|
|
|
test_comparison_utils.py
|
|
|
|
test_compile_benchmark_util.py
|
|
|
|
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
|
|
|
|
test_cpp_api_parity.py
|
Reland: Remove remaining global set_default_dtype calls from tests (#108088)
|
2023-09-07 03:04:34 +00:00 |
|
test_cpp_extensions_aot.py
|
add torch.device to python type (#108116)
|
2023-09-18 02:20:30 +00:00 |
|
test_cpp_extensions_jit.py
|
[BE]: Update ruff to 0.285 (#107519)
|
2023-08-22 23:16:38 +00:00 |
|
test_cpp_extensions_open_device_registration.py
|
|
|
|
test_cuda.py
|
[CUDA][CUDA Graphs] Fix CUDAGraph::reset function (#108896)
|
2023-09-11 19:49:31 +00:00 |
|
test_cuda_expandable_segments.py
|
|
|
|
test_cuda_multigpu.py
|
|
|
|
test_cuda_nvml_based_avail.py
|
|
|
|
test_cuda_primary_ctx.py
|
|
|
|
test_cuda_sanitizer.py
|
|
|
|
test_cuda_trace.py
|
|
|
|
test_custom_ops.py
|
[dynamo] Add BACKEND_MATCH guard to detect and recompile when backend changes (#107337)
|
2023-09-14 15:49:30 +00:00 |
|
test_dataloader.py
|
Update ruff to v0.0.286 (#108058)
|
2023-08-28 22:55:56 +00:00 |
|
test_datapipe.py
|
Disable compile for massive data pipe test (#109063)
|
2023-09-12 00:15:52 +00:00 |
|
test_decomp.py
|
Introduce Tensor overload to linspace and logspace (#104889)
|
2023-09-11 23:30:40 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
[BE]: Update ruff to 0.285 (#107519)
|
2023-08-22 23:16:38 +00:00 |
|
test_dlpack.py
|
Add DLPack bool support (#108486)
|
2023-09-08 17:55:33 +00:00 |
|
test_dynamic_shapes.py
|
remove redundant dynamic_dim (#107815)
|
2023-08-24 10:46:24 +00:00 |
|
test_expanded_weights.py
|
|
|
|
test_fake_tensor.py
|
Flash Attention v2 (#105602)
|
2023-09-13 13:59:05 +00:00 |
|
test_flop_counter.py
|
Flash Attention v2 (#105602)
|
2023-09-13 13:59:05 +00:00 |
|
test_foreach.py
|
Refactor test_foreach.py (#107869)
|
2023-09-14 19:39:26 +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.py
|
Add TorchDispatch version of functionalization (#106404)
|
2023-09-15 20:19:25 +00:00 |
|
test_functionalization_of_rng_ops.py
|
|
|
|
test_futures.py
|
|
|
|
test_fx.py
|
[dynamo] preserve some FX node metadata of GraphModules (#107067)
|
2023-09-15 23:29:14 +00:00 |
|
test_fx_experimental.py
|
Run translation validation on tracing error. (#106645)
|
2023-08-14 13:43:34 +00:00 |
|
test_fx_passes.py
|
|
|
|
test_fx_reinplace_pass.py
|
|
|
|
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
|
|
|
|
test_itt.py
|
|
|
|
test_jit.py
|
Reland: Remove remaining global set_default_dtype calls from tests (#108088)
|
2023-09-07 03:04:34 +00:00 |
|
test_jit_autocast.py
|
|
|
|
test_jit_cuda_fuser.py
|
Revert "Remove remaining global set_default_dtype calls from tests (#107246)"
|
2023-08-25 19:34:55 +00:00 |
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
[jit] Skip NNAPI, test_ivalue, CPU NNC tests in fbcode (#108937)
|
2023-09-11 22:42:30 +00:00 |
|
test_jit_legacy.py
|
|
|
|
test_jit_llga_fuser.py
|
|
|
|
test_jit_profiling.py
|
|
|
|
test_jit_simple.py
|
|
|
|
test_jit_string.py
|
|
|
|
test_jiterator.py
|
|
|
|
test_kernel_launch_checks.py
|
|
|
|
test_legacy_vmap.py
|
|
|
|
test_license.py
|
|
|
|
test_linalg.py
|
Reland: Remove remaining global set_default_dtype calls from tests (#108088)
|
2023-09-07 03:04:34 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
|
|
|
test_maskedtensor.py
|
|
|
|
test_matmul_cuda.py
|
Add better error messaging to scaled_mm (#108454)
|
2023-09-07 21:26:47 +00:00 |
|
test_meta.py
|
Add meta function for _segment_reduce (#109359)
|
2023-09-16 13:31:03 +00:00 |
|
test_metal.py
|
|
|
|
test_mkl_verbose.py
|
|
|
|
test_mkldnn.py
|
Add channels_last3d support for mkldnn conv and mkldnn deconv (#95271)
|
2023-08-30 02:53:30 +00:00 |
|
test_mkldnn_fusion.py
|
Add channels_last3d support for mkldnn conv and mkldnn deconv (#95271)
|
2023-08-30 02:53:30 +00:00 |
|
test_mkldnn_verbose.py
|
|
|
|
test_mobile_optimizer.py
|
[BE]: Update ruff to 0.285 (#107519)
|
2023-08-22 23:16:38 +00:00 |
|
test_model_dump.py
|
|
|
|
test_model_exports_to_core_aten.py
|
[aten decomp] Update sdpa decom (#108371)
|
2023-09-03 15:17:08 +00:00 |
|
test_module_init.py
|
|
|
|
test_modules.py
|
Add backward check for test_memory_format (#106104)
|
2023-08-25 18:11:54 +00:00 |
|
test_monitor.py
|
|
|
|
test_mps.py
|
Revert "Add meta and OpInfo for _embedding_bag_dense_backward (#109211)"
|
2023-09-14 22:29:12 +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
|
Add NestedTensor python subclass (#108314)
|
2023-09-11 18:29:20 +00:00 |
|
test_nn.py
|
Fix broadcasting cosine_similarity (#109363)
|
2023-09-15 17:12:35 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
Add check to prevent NumPy ndarray from being treated as tuple when indexing (#108954)
|
2023-09-15 08:51:58 +00:00 |
|
test_nvfuser_frontend.py
|
|
|
|
test_openmp.py
|
|
|
|
test_ops.py
|
Add meta function for _segment_reduce (#109359)
|
2023-09-16 13:31:03 +00:00 |
|
test_ops_fwd_gradients.py
|
Reland: Remove remaining global set_default_dtype calls from tests (#108088)
|
2023-09-07 03:04:34 +00:00 |
|
test_ops_gradients.py
|
Reland: Remove remaining global set_default_dtype calls from tests (#108088)
|
2023-09-07 03:04:34 +00:00 |
|
test_ops_jit.py
|
Reland: Remove remaining global set_default_dtype calls from tests (#108088)
|
2023-09-07 03:04:34 +00:00 |
|
test_optim.py
|
|
|
|
test_out_dtype_op.py
|
Allow registering decomps for HigherOrderOp; add decomp for out_dtype (#108080)
|
2023-08-31 03:15:38 +00:00 |
|
test_overrides.py
|
Improve torch.cuda.amp type hints (#108630)
|
2023-09-08 06:06:25 +00:00 |
|
test_package.py
|
|
|
|
test_per_overload_api.py
|
|
|
|
test_prims.py
|
|
|
|
test_proxy_tensor.py
|
Add meta function for _segment_reduce (#109359)
|
2023-09-16 13:31:03 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
|
|
|
test_python_dispatch.py
|
return_and_correct_aliasing: massage some schemas to work with torchgen (#108897)
|
2023-09-15 20:19:25 +00:00 |
|
test_pytree.py
|
Serialize pytree to json string (#106116)
|
2023-08-27 14:34:49 +00:00 |
|
test_quantization.py
|
[quant][be] Move QAT tests to its own file (#108061)
|
2023-09-15 18:34:44 +00:00 |
|
test_reductions.py
|
[BE]: Update ruff to 0.285 (#107519)
|
2023-08-22 23:16:38 +00:00 |
|
test_scatter_gather_ops.py
|
|
|
|
test_schema_check.py
|
|
|
|
test_segment_reductions.py
|
|
|
|
test_serialization.py
|
When byteorder record is missing load as little endian by default (#108343)
|
2023-09-04 15:20:22 +00:00 |
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
|
|
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
|
|
|
test_sparse.py
|
fall back to eager on NotImplementedError (#107863)
|
2023-09-07 21:22:20 +00:00 |
|
test_sparse_csr.py
|
update triton pin with needed inductor change (#107722)
|
2023-08-29 04:31:44 +00:00 |
|
test_sparse_semi_structured.py
|
[core][sparse][pruning] Add (i8i8)-> fp16 support to cuSPARSELt matmul (#109214)
|
2023-09-15 18:14:40 +00:00 |
|
test_spectral_ops.py
|
decomposition spectral ops fixes (#108360)
|
2023-09-09 04:48:09 +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
|
|
|
|
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
|
|
|
|
test_tensor_creation_ops.py
|
Reland: Remove remaining global set_default_dtype calls from tests (#108088)
|
2023-09-07 03:04:34 +00:00 |
|
test_tensorboard.py
|
Add tensorboard to pip requirements (#109349)
|
2023-09-15 10:39:48 +00:00 |
|
test_tensorexpr.py
|
|
|
|
test_tensorexpr_pybind.py
|
|
|
|
test_testing.py
|
[Inductor CUTLASS backend] Step 1: Inductor config for cuda / cutlass, util functions. (#107802)
|
2023-09-12 17:44:32 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
Add memory overlap check to meta_copy_ (#108989)
|
2023-09-12 23:28:14 +00:00 |
|
test_transformers.py
|
Flash Attention v2 (#105602)
|
2023-09-13 13:59:05 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
Reland: Remove remaining global set_default_dtype calls from tests (#108088)
|
2023-09-07 03:04:34 +00:00 |
|
test_type_promotion.py
|
|
|
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
[BE]: Update ruff to 0.285 (#107519)
|
2023-08-22 23:16:38 +00:00 |
|
test_utils.py
|
Add fast traceback utilities (#107358)
|
2023-08-18 19:05:54 +00:00 |
|
test_view_ops.py
|
Add skipLazy marker for tests and use it for tests not working with LazyTensor (#107382)
|
2023-08-22 22:34:36 +00:00 |
|
test_vulkan.py
|
|
|
|
test_weak.py
|
removing some redundant str splits (#106089)
|
2023-09-01 00:22:58 +00:00 |
|
test_xnnpack_integration.py
|
|
|
