..
ao /sparsity
[ao][sparsity] Added the Nearly Diagonal Sparsifier
2022-06-04 04:30:32 +00:00
autograd
benchmark_utils
bottleneck_test
cpp
Put symint overloads on a different name
2022-06-12 14:36:39 +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
Fix shard_module to appropriately deal with sub process groups.
2022-06-12 03:50:45 +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
Add offsets-based reduction to segment_reduce (CPU, CUDA)
2022-06-11 17:43:42 +00:00
fx
Minor FX test fix for TorchDynamo ( #79206 )
2022-06-09 20:36:04 +00:00
jit
Revert "[JIT] Propagate profiled information to DifferentiableGraph outputs"
2022-06-12 00:14:08 +00:00
jit_hooks
[ROCM] Enable custom tests on rocm
2022-04-26 14:22:14 +00:00
lazy
[LT] Add IR resuing support for manually-implemented ops
2022-05-26 16:04:47 +00:00
mobile
Revert "[mobile] Fix lightweight dispatch OOM error by introducing selective build"
2022-06-09 05:16:42 +00:00
onnx
Add onnx support for movedim and moveaxis ( #78931 )
2022-06-09 19:41:09 +00:00
package
[codemod][usort] apply import merging for fbcode (1 of 11) ( #78973 )
2022-06-06 23:44:28 +00:00
quantization
qcat: use direct memcpy when all the inputs and output share the same scale and zero_point
2022-06-10 23:59:14 +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
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
[ci] remove IN_CI env var
2022-06-11 17:16:30 +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
[forward ad] Sync conj for between primal and tangent on set forward grad
2022-06-08 04:20:17 +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
Merge torch.cuda._UntypedStorage into torch._UntypedStorage ( #75459 )
2022-05-19 13:54:39 +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
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
[test] attempt to functionalize ops with mutable positional-only args
2022-05-19 18:50:34 +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 IN_CI env var
2022-06-11 17:16:30 +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
Fix jit schema_matching ignoring self resulting in wrong operator schema
2022-06-09 19:36:06 +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
Update and improve the heuristics for linalg.lu_solve
2022-06-11 04:06:40 +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
Simplify and optimize linalg.solve
2022-06-11 04:06:40 +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
[quant][refactor] Remove the base class from __all__
2022-05-20 17:56:22 +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 layer_norm_backward ( #79189 )
2022-06-10 13:25:41 +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
Simplify and optimize linalg.solve
2022-06-11 04:06:40 +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 indexing of the underlying tensors for nested tensors ( #78934 )
2022-06-08 21:05:04 +00:00
test_nn.py
Reland: Add index value checking to MaxUnpool2d and MaxUnpool3d ( #78280 )
2022-06-03 20:09:07 +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
[ci] remove IN_CI env var
2022-06-11 17:16:30 +00:00
test_ops_gradients.py
test_ops_jit.py
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
Allow pytrees as output for make_traced and nvfuser executor ( #78802 )
2022-06-04 08:41:18 +00:00
test_profiler.py
move tree tests to the start of test_profiler.py
2022-06-12 04:24:30 +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
Autogen Tags enum, and allow specifying tags while defining an op
2022-06-11 00:29:32 +00:00
test_python_dispatch.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_pytree.py
test_quantization.py
[ao] Added fx model report per_channel detector
2022-06-10 08:09:59 +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
Add offsets-based reduction to segment_reduce (CPU, CUDA)
2022-06-11 17:43:42 +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
Support CSC/BSR/BSC inputs to unary zero-preserving functions.
2022-06-09 09:46:34 +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
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