| .. |
|
ao/sparsity
|
[ao][sparsity] Implemented state_dict() and load_state_dict() functions (#79883)
|
2022-06-24 16:55:06 +00:00 |
|
autograd
|
|
|
|
benchmark_utils
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
Fix //:module_test Conversion_MultiCUDA (#79926)
|
2022-06-21 23:32:18 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
Allow relocatable device code linking in pytorch CUDA extensions (#78225)
|
2022-06-02 21:35:56 +00:00 |
|
custom_backend
|
|
|
|
custom_operator
|
set data permits requires_grad=True on integer tensor
|
2022-06-01 15:56:32 +00:00 |
|
distributed
|
[FSDP] Fix param name prefixes for ignored modules (#79955)
|
2022-06-21 22:10:33 +00:00 |
|
distributions
|
Setting validation flag for Distributions tests to work with TorchDynamo (#80081)
|
2022-06-24 04:51:51 +00:00 |
|
error_messages
|
|
|
|
expect
|
Fix sparse BSR tensor validation.
|
2022-05-27 13:26:35 +00:00 |
|
forward_backward_compatibility
|
Make linalg_det function structured (#79486)
|
2022-06-23 23:24:44 +00:00 |
|
fx
|
Back out "Back out "[const_fold] Set requires_grad based on the folded tensor; add device_for_folding option"" (#79696)
|
2022-06-18 19:13:49 +00:00 |
|
jit
|
Use generators with all/any in torch/optim (#78142)
|
2022-06-24 17:23:45 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
Wconstab/reland pysymint (#79795)
|
2022-06-20 22:55:06 +00:00 |
|
mobile
|
Fix backward of binary_cross_entropy_with_logits
|
2022-06-23 01:31:08 +00:00 |
|
onnx
|
[ONNX Export] Use half_pixel instead of pytorch_half_pixel. (#80003)
|
2022-06-24 18:25:29 +00:00 |
|
package
|
[codemod][usort] apply import merging for fbcode (1 of 11) (#78973)
|
2022-06-06 23:44:28 +00:00 |
|
quantization
|
[quant] Add QuantizedMHA class (#79956)
|
2022-06-24 16:54:42 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
typing
|
|
|
|
allowlist_for_publicAPI.json
|
Add __all__ for torch.optim and torch.nn.modules modules (#80237)
|
2022-06-24 21:34:10 +00:00 |
|
create_dummy_torchscript_model.py
|
|
|
|
defs.bzl
|
Add all bzl files per D36874458
|
2022-06-06 09:40:19 -07:00 |
|
delete.py
|
|
|
|
HowToWriteTestsUsingFileCheck.md
|
|
|
|
linear.py
|
|
|
|
load_torchscript_model.py
|
|
|
|
run_test.py
|
Enable test: test/distributed/algorithms/ddp_comm_hooks/test_ddp_hooks.py
|
2022-06-23 00:11:53 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
[ao][sparsity] Base scheduler class for Data Schedulers (#79817)
|
2022-06-24 16:51:52 +00:00 |
|
test_autocast.py
|
|
|
|
test_autograd.py
|
Fix forward AD copy_ into same-sized tensor without fw grad
|
2022-06-17 18:55:32 +00:00 |
|
test_binary_ufuncs.py
|
Don't check overflow for scalar arg in comparison ops (#78881)
|
2022-06-05 01:43:51 +00:00 |
|
test_bundled_images.py
|
|
|
|
test_bundled_inputs.py
|
|
|
|
test_complex.py
|
|
|
|
test_cpp_api_parity.py
|
|
|
|
test_cpp_extensions_aot.py
|
Allow relocatable device code linking in pytorch CUDA extensions (#78225)
|
2022-06-02 21:35:56 +00:00 |
|
test_cpp_extensions_jit.py
|
|
|
|
test_cuda.py
|
increase sleep for TestCuda.test_caching_pinned_memory_multi_gpu (#76601)
|
2022-06-14 21:10:35 +00:00 |
|
test_cuda_primary_ctx.py
|
|
|
|
test_dataloader.py
|
Support percentages in random_split (#78877)
|
2022-06-16 02:00:25 +00:00 |
|
test_datapipe.py
|
Revert "Add validation for mapper function in datapipes with input_col (#79344)"
|
2022-06-24 17:17:33 +00:00 |
|
test_decomp.py
|
Reference implementations for rsqrt and native_layer_norm (#79413)
|
2022-06-17 07:24:02 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
Make string serialization of C++ FunctionSchema consistent with torchgen.model.FunctionSchema
|
2022-05-24 19:39:26 +00:00 |
|
test_dynamic_shapes.py
|
Revert "parse pysymints to IValues (#80066)"
|
2022-06-24 20:15:09 +00:00 |
|
test_expanded_weights.py
|
|
|
|
test_fake_tensor.py
|
[FakeTensor] Use the device of the meta tensor for fallback kernel (#80193)
|
2022-06-24 20:00:07 +00:00 |
|
test_foreach.py
|
|
|
|
test_function_schema.py
|
Expose is_out to python
|
2022-06-01 07:39:24 +00:00 |
|
test_functional_autograd_benchmark.py
|
|
|
|
test_functional_optim.py
|
|
|
|
test_functionalization.py
|
add a new alias key for functional to view op decompositions
|
2022-06-15 23:18:09 +00:00 |
|
test_futures.py
|
|
|
|
test_fx.py
|
Add Dropout1d module
|
2022-06-15 14:39:07 +00:00 |
|
test_fx_experimental.py
|
Ported proxy tensor tests over to core (#78890)
|
2022-06-07 00:28:53 +00:00 |
|
test_fx_passes.py
|
FX graph partitioner and fuser (#79439)
|
2022-06-24 18:49:37 +00:00 |
|
test_hub.py
|
|
|
|
test_import_stats.py
|
Revert "Revert "[ci] remove remaining RDS dependency""
|
2022-06-15 10:04:13 -07:00 |
|
test_indexing.py
|
fix set item to scalar tensor missing gradient info
|
2022-05-25 21:22:44 +00:00 |
|
test_jit.py
|
Revert "Enable dim=None for torch.sum (#75845)"
|
2022-06-16 22:01:41 +00:00 |
|
test_jit_autocast.py
|
Conv BN folding data type issue when conv has no bias (#78241)
|
2022-05-26 18:42:17 +00:00 |
|
test_jit_cuda_fuser.py
|
Add OpInfo for torch.equal and fix support for non-standard bools
|
2022-06-20 23:48:39 +00:00 |
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
Add OpInfo for torch.equal and fix support for non-standard bools
|
2022-06-20 23:48:39 +00:00 |
|
test_jit_legacy.py
|
Remove mentions of non-existing test_jit_py3 (#78977)
|
2022-06-07 02:28:45 +00:00 |
|
test_jit_llga_fuser.py
|
|
|
|
test_jit_profiling.py
|
Remove mentions of non-existing test_jit_py3 (#78977)
|
2022-06-07 02:28:45 +00:00 |
|
test_jit_simple.py
|
Remove mentions of non-existing test_jit_py3 (#78977)
|
2022-06-07 02:28:45 +00:00 |
|
test_jit_string.py
|
|
|
|
test_jiterator.py
|
Python Jiterator supports multiple outputs (#78139)
|
2022-05-24 21:52:56 +00:00 |
|
test_kernel_launch_checks.py
|
|
|
|
test_license.py
|
|
|
|
test_linalg.py
|
Port cholesky to structured kernels (#79300)
|
2022-06-24 02:37:45 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
|
|
|
test_meta.py
|
Make linalg_det function structured (#79486)
|
2022-06-23 23:24:44 +00:00 |
|
test_metal.py
|
|
|
|
test_mkldnn.py
|
|
|
|
test_mobile_optimizer.py
|
|
|
|
test_model_dump.py
|
|
|
|
test_module_init.py
|
[quant] Add QuantizedMHA class (#79956)
|
2022-06-24 16:54:42 +00:00 |
|
test_modules.py
|
Support both train / eval modes for ModuleInfo
|
2022-06-09 20:57:17 +00:00 |
|
test_monitor.py
|
|
|
|
test_mps.py
|
[MPS] Add L1 loss test (#80010)
|
2022-06-24 17:18:31 +00:00 |
|
test_multiprocessing.py
|
Fix MPS interaction with autograd engine
|
2022-05-17 21:26:16 +00:00 |
|
test_multiprocessing_spawn.py
|
|
|
|
test_namedtensor.py
|
Revert "Enable dim=None for torch.sum (#75845)"
|
2022-06-16 22:01:41 +00:00 |
|
test_namedtuple_return_api.py
|
Make linalg_det function structured (#79486)
|
2022-06-23 23:24:44 +00:00 |
|
test_native_functions.py
|
|
|
|
test_native_mha.py
|
[PyTorch] Round T up to next multiple of 8 in NestedTensor case
|
2022-05-25 20:34:19 +00:00 |
|
test_nestedtensor.py
|
Support softmax(nested tensor) (#80179)
|
2022-06-24 14:50:24 +00:00 |
|
test_nn.py
|
BUG: Evade segfault by throwing a RuntimeError for nn.ChannelShuffle and empty input tensors (#77029)
|
2022-06-23 21:14:02 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
ENH: Add a force argument to numpy() (#78564)
|
2022-06-06 14:14:17 +00:00 |
|
test_openmp.py
|
|
|
|
test_ops.py
|
Enable torch._prims.amax/amin for nvFuser executor (#80070)
|
2022-06-23 10:19:57 +00:00 |
|
test_ops_gradients.py
|
|
|
|
test_ops_jit.py
|
Minor fix in jit tests to pass TorchDynamo (#79903)
|
2022-06-21 00:42:02 +00:00 |
|
test_optim.py
|
Fix SequentialLR initialization (#72856)
|
2022-06-21 20:21:13 +00:00 |
|
test_overrides.py
|
add utils for checking that all modes are in the same scope and finding the outermost mode
|
2022-06-10 19:31:05 +00:00 |
|
test_package.py
|
|
|
|
test_per_overload_api.py
|
|
|
|
test_prims.py
|
small cleanup of executor (#79973)
|
2022-06-22 00:35:51 +00:00 |
|
test_profiler.py
|
[Profiler] Add queue depth computation (#79993)
|
2022-06-24 20:47:58 +00:00 |
|
test_profiler_tree.py
|
[Profiler][Trivial] Update test_profiler_tree
|
2022-06-22 22:30:33 +00:00 |
|
test_proxy_tensor.py
|
Switched to tracing overloads by default (#80013)
|
2022-06-22 18:55:06 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
Wconstab/reland pysymint (#79795)
|
2022-06-20 22:55:06 +00:00 |
|
test_python_dispatch.py
|
disable modes during constructor
|
2022-06-17 22:28:27 +00:00 |
|
test_pytree.py
|
|
|
|
test_quantization.py
|
[ao] Adds framework for InputWeightEqualization Detector (#79916)
|
2022-06-24 14:51:15 +00:00 |
|
test_reductions.py
|
Add complex_to_float option in ReductionOpInfo
|
2022-06-01 09:31:45 +00:00 |
|
test_scatter_gather_ops.py
|
fix gather sparse_grad backward crash with empty index tensor
|
2022-06-04 02:25:05 +00:00 |
|
test_segment_reductions.py
|
reland Add offsets-based reduction to segment_reduce (CPU, CUDA)
|
2022-06-17 15:49:31 +00:00 |
|
test_serialization.py
|
|
|
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
|
|
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
MAINT: Harmonize argsort params with array_api (#75162)
|
2022-06-09 12:32:01 +00:00 |
|
test_sparse.py
|
index_select for COO CUDA tensors. (#77551)
|
2022-06-01 17:39:03 +00:00 |
|
test_sparse_csr.py
|
optimize to_dense for CSC (#79635)
|
2022-06-21 16:52:16 +00:00 |
|
test_spectral_ops.py
|
|
|
|
test_stateless.py
|
Added setattr to functional_call. (#77137)
|
2022-05-17 05:40:46 +00:00 |
|
test_static_runtime.py
|
[Static Runtime] Added nested prim::fork and aten::wait test case (#79746)
|
2022-06-20 16:32:17 +00:00 |
|
test_subclass.py
|
Throw a nice error when SubTensor.__torch_dispatch__() returns the wrong type for detach()
|
2022-05-18 20:00:42 +00:00 |
|
test_tensor_creation_ops.py
|
|
|
|
test_tensorboard.py
|
[tensorboard] update assertion error for scalar() and fix docs (#76859)
|
2022-06-16 05:24:20 +00:00 |
|
test_tensorexpr.py
|
[NNC] channels last propagation within NNC fusion group (#76948)
|
2022-05-30 18:31:49 +00:00 |
|
test_tensorexpr_pybind.py
|
|
|
|
test_testing.py
|
[fx2trt] move common_fx2trt.py into fx folder (#79924)
|
2022-06-22 17:54:10 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
More forward AD formulas
|
2022-06-23 01:31:08 +00:00 |
|
test_transformers.py
|
Add test for torchscripting nn.TransformerEncoder, including fast path (#79796)
|
2022-06-17 22:00:49 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
ENH: Convert finfo.tiny to finfo.smallest_normal (#76292)
|
2022-05-20 00:59:48 +00:00 |
|
test_type_promotion.py
|
[fix] complex type promotion (#77524)
|
2022-05-20 10:23:56 +00:00 |
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
update signbit docs and add -0. to reference testing for unary and binary functions. (#78349)
|
2022-06-06 13:48:08 +00:00 |
|
test_utils.py
|
Revert "To add hipify_torch as a submodule in pytorch/third_party (#74704)"
|
2022-06-21 23:54:00 +00:00 |
|
test_view_ops.py
|
[primTorch] Adds broadcast_shapes reference (#78612)
|
2022-06-02 08:56:37 +00:00 |
|
test_vmap.py
|
|
|
|
test_vulkan.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
