| .. |
|
ao/sparsity
|
[ao][sparsity] Added the Nearly Diagonal Sparsifier
|
2022-06-04 04:30:32 +00:00 |
|
autograd
|
Fix autograd.functional tests to not fail with logging tensor
|
2022-04-20 20:32:40 +00:00 |
|
benchmark_utils
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
have a common interface to extract metadata from SizeNodes (#78088)
|
2022-06-15 04:59:08 +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
|
[ROCM] Enable custom tests on rocm
|
2022-04-26 14:22:14 +00:00 |
|
custom_operator
|
set data permits requires_grad=True on integer tensor
|
2022-06-01 15:56:32 +00:00 |
|
distributed
|
[PT-D] Use process group of the partial tensor so sub pg comm will be enabled during reshard
|
2022-06-14 17:44:51 +00:00 |
|
distributions
|
Add mode property to distributions. (#76690)
|
2022-05-11 18:26:56 +00:00 |
|
error_messages
|
|
|
|
expect
|
Fix sparse BSR tensor validation.
|
2022-05-27 13:26:35 +00:00 |
|
forward_backward_compatibility
|
Fix BC Test for SymInt (#79612)
|
2022-06-15 15:54:40 +00:00 |
|
fx
|
[const_fold] Set requires_grad based on the folded tensor; add device_for_folding option (#79067)
|
2022-06-14 19:00:05 +00:00 |
|
jit
|
Retry - [JIT] Add mutation checks for tensor inputs
|
2022-06-13 18:16:50 +00:00 |
|
jit_hooks
|
[ROCM] Enable custom tests on rocm
|
2022-04-26 14:22:14 +00:00 |
|
lazy
|
Revert "Python Bindings for SymInts (#78135)"
|
2022-06-15 13:52:14 +00:00 |
|
mobile
|
[ci] don't explicitly set CUSTOM_TEST_ARTIFACT_BUILD_DIR
|
2022-06-13 16:09:50 +00:00 |
|
onnx
|
[ONNX] subscribe onnx to our custom test infra (#79546)
|
2022-06-15 15:00:04 +00:00 |
|
package
|
[codemod][usort] apply import merging for fbcode (1 of 11) (#78973)
|
2022-06-06 23:44:28 +00:00 |
|
quantization
|
[ao] Added function to inform dynamic vs static appropriate
|
2022-06-15 02:51:27 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
typing
|
|
|
|
allowlist_for_publicAPI.json
|
Simplify and optimize linalg.solve
|
2022-06-11 04:06:40 +00:00 |
|
create_dummy_torchscript_model.py
|
Add test for FC/BC for torchscript file.
|
2022-04-13 23:23:13 +00:00 |
|
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
|
Add test for FC/BC for torchscript file.
|
2022-04-13 23:23:13 +00:00 |
|
run_test.py
|
[ci] remove TD + test specification code from run_test.py
|
2022-06-13 16:09:53 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
[ao][sparsity] Added the Nearly Diagonal Sparsifier
|
2022-06-04 04:30:32 +00:00 |
|
test_autocast.py
|
Update operator list for AutocastCPU (#68725)
|
2022-05-11 17:28:35 +00:00 |
|
test_autograd.py
|
[WIP] Fix non-reentrant hooks based checkpointing
|
2022-06-14 01:13:33 +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
|
Remove breakpad dependency
|
2022-05-03 20:21:55 +00:00 |
|
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
|
[ci] remove IN_CI env var
|
2022-06-11 17:16:30 +00:00 |
|
test_datapipe.py
|
[DataPipe] Adding functional API for FileLister (#78419)
|
2022-06-06 17:26:19 +00:00 |
|
test_decomp.py
|
Minor updates for upcast_tensor
|
2022-06-01 00:46:52 +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_expanded_weights.py
|
expanded weights: instance norm faster rule
|
2022-04-19 19:40:09 +00:00 |
|
test_fake_tensor.py
|
Add Dynamic Output Shape Tagdfor ata-dependent ops, handle in FakeTensor
|
2022-06-09 22:16:16 +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
|
Revert "fix _unsafe_view schema to work with functionalization"
|
2022-06-10 13:09:00 +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_hub.py
|
|
|
|
test_import_stats.py
|
[ci] remove remaining RDS dependency
|
2022-06-13 18:46:04 +00:00 |
|
test_indexing.py
|
fix set item to scalar tensor missing gradient info
|
2022-05-25 21:22:44 +00:00 |
|
test_jit.py
|
[jit] fix indexing into a tensor with a tuple
|
2022-06-13 19:51:47 +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
|
Revert "[nvfuser_upstream_push] nvfuser code base bump 060822 (#79147)"
|
2022-06-10 20:55:10 +00:00 |
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
Disable TracerWarnings on NNC opinfo tests
|
2022-06-03 18:11:12 +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
|
[Reland take-2] Add JIT graph fuser for oneDNN Graph API (v0.5)
|
2022-05-05 16:57:03 +00:00 |
|
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 linalg_eigh and linalg_eigvalsh to structured
|
2022-06-14 20:17:01 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
Use scatter_reduce to support masked reductions on sparse COO tensors (sum, prod, amin, amax)
|
2022-05-06 15:40:22 +00:00 |
|
test_meta.py
|
Port linalg_eigh and linalg_eigvalsh to structured
|
2022-06-14 20:17:01 +00:00 |
|
test_metal.py
|
|
|
|
test_mkldnn.py
|
Enable fp32/bf16 PRelu forward and backward in MkldnnCPU path (#60427)
|
2022-05-10 17:29:11 +00:00 |
|
test_mobile_optimizer.py
|
|
|
|
test_model_dump.py
|
|
|
|
test_module_init.py
|
Add Dropout1d module
|
2022-06-15 14:39:07 +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] Support stride of stride
|
2022-06-14 18:49:44 +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
|
|
|
|
test_namedtuple_return_api.py
|
Port linalg_eigh and linalg_eigvalsh to structured
|
2022-06-14 20:17:01 +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
|
Revert "Python Bindings for SymInts (#78135)"
|
2022-06-15 13:52:14 +00:00 |
|
test_nn.py
|
Revert behavior of Dropout2d on 3D inputs to 1D channel-wise dropout behavior & warn
|
2022-06-15 14:56:43 +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
|
[chalf] enable skipped tests (#79376)
|
2022-06-13 17:31:45 +00:00 |
|
test_ops_gradients.py
|
Various OpInfo architecture improvements
|
2022-04-18 21:55:32 +00:00 |
|
test_ops_jit.py
|
Various OpInfo architecture improvements
|
2022-04-18 21:55:32 +00:00 |
|
test_optim.py
|
Adding maximize to Adamax (#77409)
|
2022-05-16 17:34:44 +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
|
Enable torch._refs.var for nvFuser executor (#79517)
|
2022-06-14 23:08:53 +00:00 |
|
test_profiler.py
|
Use custom AppendOnlyList for op_events to reduce the number of atomic operations (#78643)
|
2022-06-14 03:17:09 +00:00 |
|
test_profiler_tree.py
|
move tree tests to the start of test_profiler.py
|
2022-06-12 04:24:30 +00:00 |
|
test_proxy_tensor.py
|
Ported proxy tensor tests over to core (#78890)
|
2022-06-07 00:28:53 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
Revert "Python Bindings for SymInts (#78135)"
|
2022-06-15 13:52:14 +00:00 |
|
test_python_dispatch.py
|
add sizes to slowpath
|
2022-06-14 01:19:59 +00:00 |
|
test_pytree.py
|
Register torch.return_types.* as pytree nodes
|
2022-04-19 13:46:20 +00:00 |
|
test_quantization.py
|
[ao] Added function to inform dynamic vs static appropriate
|
2022-06-15 02:51:27 +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
|
Revert "Add offsets-based reduction to segment_reduce (CPU, CUDA)"
|
2022-06-13 22:37:25 +00:00 |
|
test_serialization.py
|
Make sure requires_grad is propagated for all backend
|
2022-04-25 19:31:24 +00:00 |
|
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
|
[ROCm] Enable some sparse tests on ROCm (#77877)
|
2022-06-14 21:11:35 +00:00 |
|
test_spectral_ops.py
|
[complex32] fft support (cuda only) (#74857)
|
2022-05-12 04:28:55 +00:00 |
|
test_stateless.py
|
Added setattr to functional_call. (#77137)
|
2022-05-17 05:40:46 +00:00 |
|
test_static_runtime.py
|
[Static Runtime] Exception handling during fork subgraph execution (#79292)
|
2022-06-11 03:11:49 +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
|
Add BFloat16 support for GLU, and randperm operators on CPU (#61944)
|
2022-05-12 17:41:57 +00:00 |
|
test_tensorboard.py
|
Cannibalize noarch CI job into crossref CI job
|
2022-04-20 11:56:25 +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
|
[NNC] Lowering function generates the output buffer with the specified stride (#76529)
|
2022-05-04 20:04:22 +00:00 |
|
test_testing.py
|
[ci] remove IN_CI env var
|
2022-06-11 17:16:30 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
update is_same_size to work with nested tensor dispatch
|
2022-06-11 00:07:27 +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
|
Fix issue with _checkpoint_without_reentrant
|
2022-05-05 17:37:31 +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
|
|
|