| .. |
|
_nvfuser
|
adding symbolic link to get CI to run tests where cmake is not run on CI node (#95402)
|
2023-03-01 19:01:18 +00:00 |
|
ao/sparsity
|
Back out "[core][pruning][be] rename BaseSparsifier to BasePruner (#98747)" (#99171)
|
2023-04-15 00:37:45 +00:00 |
|
autograd
|
|
|
|
backends/xeon
|
enable taskset core pinning in addition to numactl (#96011)
|
2023-03-07 01:19:46 +00:00 |
|
benchmark_utils
|
[BE] Remove unnecessary dict comprehensions (#97116)
|
2023-03-20 00:56:57 +00:00 |
|
bottleneck_test
|
|
|
|
cpp
|
[reland][BE][autograd Function] Raise an error if input is returned a… (#98051)
|
2023-04-11 15:42:54 +00:00 |
|
cpp_api_parity
|
[BE] Remove unnecessary dict comprehensions (#97116)
|
2023-03-20 00:56:57 +00:00 |
|
cpp_extensions
|
Automatically generate attributes and methods for custom backends. (#98066)
|
2023-04-13 22:04:05 +00:00 |
|
custom_backend
|
|
|
|
custom_operator
|
Enable TestTorchbind on Windows (#96507)
|
2023-03-16 16:18:08 +00:00 |
|
distributed
|
Enable FSDP `use_orig_params=True` mixed precision training when some ranks have no (non-zero sized) parameter shards (#99175)
|
2023-04-15 05:13:23 +00:00 |
|
distributions
|
Fix gumbel cdf (#91698)
|
2023-03-07 23:04:47 +00:00 |
|
dynamo
|
[Dynamo] No graph break for explicit calling Conv{1/2/3}d.forward & ConvTranspose{1/2/3}d.forward (#99015)
|
2023-04-15 08:04:13 +00:00 |
|
edge
|
|
|
|
error_messages
|
|
|
|
expect
|
[PATCH] Back out "Move functional collectives implementation to python. (#98595) (#99168)
|
2023-04-14 23:48:19 +00:00 |
|
export
|
[export] Constraints API (#98433)
|
2023-04-13 21:20:10 +00:00 |
|
forward_backward_compatibility
|
[PATCH] Back out "Move functional collectives implementation to python. (#98595) (#99168)
|
2023-04-14 23:48:19 +00:00 |
|
functorch
|
Add vmap support for torch.nn.functional.smooth_l1_loss (#98357)
|
2023-04-14 12:29:53 +00:00 |
|
fx
|
Update InternalMatch in subgraph_rewriter after repeated replacements (#99039)
|
2023-04-14 19:35:38 +00:00 |
|
inductor
|
Unbreak torch.compile on macos (#99119)
|
2023-04-14 15:30:27 +00:00 |
|
jit
|
[BE] Update flake8-comprehensions and adapt to rule C418 (#99178)
|
2023-04-15 15:33:42 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
|
|
|
mobile
|
|
|
|
nn
|
Fix module backward pre-hooks to actually update gradient (#97983)
|
2023-03-30 20:33:44 +00:00 |
|
onnx
|
[ONNX] Support aten.stack for dynamo_export (#99191)
|
2023-04-15 01:13:00 +00:00 |
|
onnx_caffe2
|
|
|
|
package
|
[torch package][easy] Make all the save/load tests use buffers (#98798)
|
2023-04-14 13:52:17 +00:00 |
|
profiler
|
Removed hip call hipDeviceSynchronize (#97209)
|
2023-04-09 20:12:52 +00:00 |
|
quantization
|
[quant][fix] Compare resnet with quantizer api with the prepare_fx and decomposed convert flow (#98905)
|
2023-04-14 16:25:15 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
typing
|
|
|
|
allowlist_for_publicAPI.json
|
|
|
|
conftest.py
|
Fix pytest config (#98607)
|
2023-04-08 00:55:51 +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 |
|
mkl_verbose.py
|
|
|
|
mkldnn_verbose.py
|
|
|
|
run_doctests.sh
|
|
|
|
run_test.py
|
Expandable blocks in allocator (#96995)
|
2023-04-14 09:49:11 +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
|
Back out "[core][pruning][be] rename BaseSparsifier to BasePruner (#98747)" (#99171)
|
2023-04-15 00:37:45 +00:00 |
|
test_autocast.py
|
|
|
|
test_autograd.py
|
[BE] Enable flake8-comprehension rule C417 (#97880)
|
2023-03-30 14:34:24 +00:00 |
|
test_binary_ufuncs.py
|
[BE] Enable flake8-comprehension rule C417 (#97880)
|
2023-03-30 14:34:24 +00:00 |
|
test_bundled_images.py
|
|
|
|
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
|
torch.compile benchmark utility (#97699)
|
2023-04-12 03:02:06 +00:00 |
|
test_complex.py
|
Fix CPU vectorized eq and ne operations for complex types (#97374)
|
2023-04-14 02:02:16 +00:00 |
|
test_cpp_api_parity.py
|
|
|
|
test_cpp_extensions_aot.py
|
|
|
|
test_cpp_extensions_jit.py
|
|
|
|
test_cpp_extensions_open_device_registration.py
|
Automatically generate attributes and methods for custom backends. (#98066)
|
2023-04-13 22:04:05 +00:00 |
|
test_cuda.py
|
Expandable blocks in allocator (#96995)
|
2023-04-14 09:49:11 +00:00 |
|
test_cuda_expandable_segments.py
|
Expandable blocks in allocator (#96995)
|
2023-04-14 09:49:11 +00:00 |
|
test_cuda_nvml_based_avail.py
|
Reduce pytest blocklist (#96016)
|
2023-03-07 18:30:27 +00:00 |
|
test_cuda_primary_ctx.py
|
Reduce pytest blocklist (#96016)
|
2023-03-07 18:30:27 +00:00 |
|
test_cuda_sanitizer.py
|
Reduce pytest blocklist (#96016)
|
2023-03-07 18:30:27 +00:00 |
|
test_cuda_trace.py
|
Reduce pytest blocklist (#96016)
|
2023-03-07 18:30:27 +00:00 |
|
test_dataloader.py
|
Expandable blocks in allocator (#96995)
|
2023-04-14 09:49:11 +00:00 |
|
test_datapipe.py
|
[BE] Update flake8-comprehensions and adapt to rule C418 (#99178)
|
2023-04-15 15:33:42 +00:00 |
|
test_decomp.py
|
addmv decomp #2 (#96264)
|
2023-03-16 23:09:45 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
|
|
|
test_dlpack.py
|
|
|
|
test_dynamic_shapes.py
|
Refactor dynamic dims api, stateless internals, higher level export API (#96699)
|
2023-03-29 16:55:49 +00:00 |
|
test_expanded_weights.py
|
|
|
|
test_fake_tensor.py
|
Recompute flat_arg_fake_tensors after fakification (#98769)
|
2023-04-14 19:14:29 +00:00 |
|
test_flop_counter.py
|
fix flop formulas for sdpa (#96690)
|
2023-03-16 04:55:56 +00:00 |
|
test_foreach.py
|
[test_foreach] add cases of zero size tensors (#95028)
|
2023-03-23 00:12:13 +00:00 |
|
test_function_schema.py
|
|
|
|
test_functional_autograd_benchmark.py
|
|
|
|
test_functional_optim.py
|
|
|
|
test_functionalization.py
|
dont bake in defaults when tracing *_like factories (#97564)
|
2023-03-27 22:53:44 +00:00 |
|
test_futures.py
|
|
|
|
test_fx.py
|
Fix typos in torch/fx/_compatibility.py (#97618)
|
2023-03-29 21:55:13 +00:00 |
|
test_fx_experimental.py
|
|
|
|
test_fx_passes.py
|
|
|
|
test_fx_reinplace_pass.py
|
|
|
|
test_hub.py
|
torch.hub: add safe weights_only option to load_state_dict_from_url (#98479)
|
2023-04-11 12:44:25 +00:00 |
|
test_import_stats.py
|
|
|
|
test_indexing.py
|
|
|
|
test_itt.py
|
|
|
|
test_jit.py
|
[PyTorch] Optimize DictType::annotation_str_impl (#96498)
|
2023-03-24 02:38:21 +00:00 |
|
test_jit_autocast.py
|
[Dynamo] Support torch.{cuda/cpu}.amp.autocast (#95416)
|
2023-03-10 21:48:08 +00:00 |
|
test_jit_cuda_fuser.py
|
adding symbolic link to get CI to run tests where cmake is not run on CI node (#95402)
|
2023-03-01 19:01:18 +00:00 |
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
Skip test_batch_norm in test_jit_fuser_te for asan (#98016)
|
2023-03-30 20:58:41 +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
|
Reduce pytest blocklist (#96016)
|
2023-03-07 18:30:27 +00:00 |
|
test_kernel_launch_checks.py
|
|
|
|
test_legacy_vmap.py
|
|
|
|
test_license.py
|
|
|
|
test_linalg.py
|
[CUDA][CUDA 11] Remove more CUDA 11 version checks (#92934)
|
2023-03-30 19:49:52 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
|
|
|
test_maskedtensor.py
|
fix random mask creation in test_maskedtensor (#97017)
|
2023-03-24 23:55:17 +00:00 |
|
test_matmul_cuda.py
|
Enable test_matmul_cuda UTs for ROCm (#98797)
|
2023-04-13 19:36:07 +00:00 |
|
test_meta.py
|
[primTorch] Add count_nonzero (#98995)
|
2023-04-13 22:08:19 +00:00 |
|
test_metal.py
|
|
|
|
test_mkl_verbose.py
|
|
|
|
test_mkldnn.py
|
Split rnn primitive for inference and training (#96736)
|
2023-03-27 11:14:15 +00:00 |
|
test_mkldnn_fusion.py
|
|
|
|
test_mkldnn_verbose.py
|
|
|
|
test_mobile_optimizer.py
|
|
|
|
test_model_dump.py
|
|
|
|
test_module_init.py
|
Adding nn.ZeroPad1d and nn.ZeroPad3d (#96295)
|
2023-03-10 03:51:41 +00:00 |
|
test_modules.py
|
Added ModuleInfos for Pooling ops (#98358)
|
2023-04-05 19:39:07 +00:00 |
|
test_monitor.py
|
|
|
|
test_mps.py
|
Fix test_mps for macos 13.3 (#98739)
|
2023-04-12 19:23:08 +00:00 |
|
test_multiprocessing.py
|
Expandable blocks in allocator (#96995)
|
2023-04-14 09:49:11 +00:00 |
|
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 ops: logical_not, logical_not_, masked_fill (#97934)
|
2023-03-30 08:14:39 +00:00 |
|
test_nn.py
|
Updated upsampling test to use parametrize_test decorator (#97769)
|
2023-04-11 12:20:00 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
|
|
|
test_nvfuser_dynamo.py
|
adding symbolic link to get CI to run tests where cmake is not run on CI node (#95402)
|
2023-03-01 19:01:18 +00:00 |
|
test_nvfuser_frontend.py
|
adding symbolic link to get CI to run tests where cmake is not run on CI node (#95402)
|
2023-03-01 19:01:18 +00:00 |
|
test_openmp.py
|
|
|
|
test_ops.py
|
Enable fake_crossref unit tests on rocm (#97368)
|
2023-04-12 02:38:35 +00:00 |
|
test_ops_fwd_gradients.py
|
|
|
|
test_ops_gradients.py
|
|
|
|
test_ops_jit.py
|
|
|
|
test_optim.py
|
Change 1D Tensor of 1 element to 0D Tensor (#96994)
|
2023-03-21 18:24:19 +00:00 |
|
test_overrides.py
|
Make gen_annotated_args support kwargs (#98396)
|
2023-04-06 19:42:26 +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
|
|
|
|
test_proxy_tensor.py
|
[primTorch] Add count_nonzero (#98995)
|
2023-04-13 22:08:19 +00:00 |
|
test_pruning_op.py
|
Add missing __main__ in two unittests (#97302)
|
2023-03-22 19:09:08 +00:00 |
|
test_public_bindings.py
|
Add Symbool support in python to C++ translation (#98453)
|
2023-04-12 03:21:57 +00:00 |
|
test_python_dispatch.py
|
[torch.library] Add ability to create library fragments (#98439)
|
2023-04-10 18:04:53 +00:00 |
|
test_pytree.py
|
|
|
|
test_quantization.py
|
Init quantization backend config for inductor (#96476)
|
2023-03-22 07:56:56 +00:00 |
|
test_reductions.py
|
|
|
|
test_scatter_gather_ops.py
|
add deterministic impl for scatter and scatter_reduction sum/mean mode (#98060)
|
2023-04-03 20:38:29 +00:00 |
|
test_schema_check.py
|
|
|
|
test_segment_reductions.py
|
|
|
|
test_serialization.py
|
Add test for pickle_module (#98373)
|
2023-04-05 13:05:05 +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
|
Enable cadd_sparse for BFloat16 on CPU (#96767)
|
2023-04-14 19:50:49 +00:00 |
|
test_sparse_csr.py
|
nn.Linear: dispatch to bsr_dense_mm for half and bfloat16 (#94825)
|
2023-04-15 13:38:42 +00:00 |
|
test_spectral_ops.py
|
Fix usages of contextmanager without finally (#96170)
|
2023-03-08 20:59:27 +00:00 |
|
test_stateless.py
|
|
|
|
test_static_runtime.py
|
|
|
|
test_subclass.py
|
|
|
|
test_sympy_utils.py
|
Use maxint to bound integers. (#96121)
|
2023-03-07 12:46:19 +00:00 |
|
test_tensor_creation_ops.py
|
Add overflow check for stride calculation (#94900)
|
2023-04-09 01:30:55 +00:00 |
|
test_tensorboard.py
|
Use is_available instead of device_count to check for CUDA availability (#97043)
|
2023-03-18 00:39:42 +00:00 |
|
test_tensorexpr.py
|
|
|
|
test_tensorexpr_pybind.py
|
|
|
|
test_testing.py
|
[ROCm] Enable test_filtering_env_var (#84100)
|
2023-04-04 21:49:53 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
Add itemsize and nbytes properties to Tensor (#98322)
|
2023-04-05 12:11:55 +00:00 |
|
test_transformers.py
|
Adds 3D attn_mask support to merge_masks() for Multihead Attention fast path (#98991)
|
2023-04-13 20:29:57 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
|
|
|
test_type_promotion.py
|
[BE] Enable flake8-comprehension rule C417 (#97880)
|
2023-03-30 14:34:24 +00:00 |
|
test_typing.py
|
[BE] Enable flake8-comprehension rule C417 (#97880)
|
2023-03-30 14:34:24 +00:00 |
|
test_unary_ufuncs.py
|
|
|
|
test_utils.py
|
add get_device_index for custom device (#98804)
|
2023-04-12 23:58:31 +00:00 |
|
test_view_ops.py
|
Add overflow check for stride calculation (#94900)
|
2023-04-09 01:30:55 +00:00 |
|
test_vulkan.py
|
|
|
|
test_weak.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
