pytorch

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

Add CPU slow test job (#73748 )

2022-03-23 21:17:27 +00:00

bottleneck_test

Set test owners for tests with unknown owners (#67552 )

2021-10-29 12:42:01 -07:00

cpp

turn on -Werror=unused-function in our Bazel CPU build

2022-06-10 22:11:54 +00:00

cpp_api_parity

ENH Adds test and docs for modules that already support no batch dims (#62729 )

2021-09-02 12:36:54 -07:00

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

Checkpoint util

2022-06-10 18:37:36 +00:00

distributions

Add mode property to distributions. (#76690 )

2022-05-11 18:26:56 +00:00

error_messages

Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default

2021-08-12 11:45:01 -07:00

expect

Fix sparse BSR tensor validation.

2022-05-27 13:26:35 +00:00

forward_backward_compatibility

Revert "fix _unsafe_view schema to work with functionalization"

2022-06-10 13:09:00 +00:00

fx

Minor FX test fix for TorchDynamo (#79206 )

2022-06-09 20:36:04 +00:00

jit

Revert "Add mutation checks for tensor inputs"

2022-06-10 20:15:30 +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

Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default

2021-08-12 11:45:01 -07:00

test_img

[PyTorch Operator] [2/n] Adding python test

2020-06-25 18:36:44 -07:00

typing

Add typing for torch.return_type

2022-03-29 02:17:21 +00:00

allowlist_for_publicAPI.json

Add linalg.lu_solve

2022-06-07 22:28:28 +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

Introducing DataChunk for DataPipes batching (#62768 )

2021-08-06 08:38:33 -07:00

HowToWriteTestsUsingFileCheck.md

Allow for heterogenous List and Dict values + Improve container typing algorithm (#57137 )

2021-07-10 14:29:05 -07:00

linear.py

Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default

2021-08-12 11:45:01 -07:00

load_torchscript_model.py

Add test for FC/BC for torchscript file.

2022-04-13 23:23:13 +00:00

run_test.py

[ROCm] Enable test_instantiator, test_type_hints (#78633 )

2022-06-06 06:09:34 +00:00

simulate_nccl_errors.py

Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default

2021-08-12 11:45:01 -07:00

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

Add ownership to more edge tests (#67859 )

2021-11-05 11:01:16 -07:00

test_bundled_inputs.py

Add lint to ensure all test files have headers with ownership info (#66826 )

2021-11-03 18:21:49 -07:00

test_complex.py

[testing] Update dispatch macros (#74977 )

2022-03-30 14:13:21 -07:00

test_cpp_api_parity.py

[skip ci] Set test owners for cpp test (#66836 )

2021-10-21 13:17:46 -07:00

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_primary_ctx.py

remove hasPrimaryContext workaround on ROCm (#71146 )

2022-01-25 20:32:12 +00:00

test_cuda.py

Merge torch.cuda._UntypedStorage into torch._UntypedStorage (#75459 )

2022-05-19 13:54:39 +00:00

test_dataloader.py

[DataLoader] DataLoader now automatically apply sharding to DataPipes

2022-06-02 17:40:29 +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

[skip ci] Set test owner for deploy/package tests (#66830 )

2021-10-26 07:49:33 -07:00

test_determination.py

Have test classes extend from common_utils.TestCase, not unittest.TestCase (#66900 )

2021-10-19 16:54:05 -07:00

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

[testing] Update dispatch macros (#74977 )

2022-03-30 14:13:21 -07:00

test_function_schema.py

Expose is_out to python

2022-06-01 07:39:24 +00:00

test_functional_autograd_benchmark.py

fix python version error (#71021 )

2022-01-21 01:05:53 +00:00

test_functional_optim.py

[Opt Overlap] Clean up code in _OptimizerHookState (#71620 )

2022-01-26 19:33:49 +00:00

test_functionalization.py

Revert "fix _unsafe_view schema to work with functionalization"

2022-06-10 13:09:00 +00:00

test_futures.py

Set test owners for tests with unknown owners (#67552 )

2021-10-29 12:42:01 -07:00

test_fx_experimental.py

Ported proxy tensor tests over to core (#78890 )

2022-06-07 00:28:53 +00:00

test_fx.py

[test] attempt to functionalize ops with mutable positional-only args

2022-05-19 18:50:34 +00:00

test_hub.py

torch.hub security improvement: add new trust_repo parameter

2022-04-05 09:29:25 +00:00

test_import_stats.py

Set test owners for module: ci (#66796 )

2021-10-18 10:29:50 -07:00

test_indexing.py

fix set item to scalar tensor missing gradient info

2022-05-25 21:22:44 +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

Set test owners for jit tests (#66808 )

2021-10-25 07:51:10 -07:00

test_jit_fuser_legacy.py

Set test owners for jit tests (#66808 )

2021-10-25 07:51:10 -07:00

test_jit_fuser_te.py

Disable TracerWarnings on NNC opinfo tests

2022-06-03 18:11:12 +00:00

test_jit_fuser.py

skip test_jit_legacy (#68129 )

2021-11-10 23:08:32 -08: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

[PyTorch] Move prim string ops to JIT op registry (#70501 )

2022-01-06 18:26:28 -08:00

test_jit.py

Fix jit schema_matching ignoring self resulting in wrong operator schema

2022-06-09 19:36:06 +00:00

test_jiterator.py

Python Jiterator supports multiple outputs (#78139 )

2022-05-24 21:52:56 +00:00

test_kernel_launch_checks.py

Set test owners for tests with unknown owners (#67552 )

2021-10-29 12:42:01 -07:00

test_license.py

Set test owners for tests with unknown owners (#67552 )

2021-10-29 12:42:01 -07:00

test_linalg.py

Port linalg_qr to structured

2022-06-09 14:41:30 +00:00

test_logging.py

[Codemod][Codemod deprecated unittest asserts] fbcode//caffe2/test (#71708 )

2022-03-15 19:28:52 +00:00

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

[meta] angle, angle.out (#79278 )

2022-06-10 20:06:31 +00:00

test_metal.py

[skip ci] Set test owner for mobile tests (#66829 )

2021-10-26 10:20:01 -07:00

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

Prepare for an update to the XNNPACK submodule (#72642 )

2022-02-25 00:39:15 +00:00

test_model_dump.py

Prepare for an update to the XNNPACK submodule (#72642 )

2022-02-25 00:39:15 +00:00

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

torch/monitor: merge Interval and FixedCount stats (#72009 )

2022-01-30 23:21:59 +00:00

test_mps.py

MPS: add layer_norm_backward (#79189 )

2022-06-10 13:25:41 +00:00

test_multiprocessing_spawn.py

Make test_multiprocessing_spawn.py compatible with pytest (#50408 )

2022-02-10 16:18:46 +00:00

test_multiprocessing.py

Fix MPS interaction with autograd engine

2022-05-17 21:26:16 +00:00

test_namedtensor.py

use torch.testing.assert_equal in TestCase.assertEqual (#67796 )

2022-01-27 08:33:55 +00:00

test_namedtuple_return_api.py

Remove deprecated torch.solve (#70986 )

2022-05-10 13:44:07 +00:00

test_native_functions.py

Set test owners for tests with unknown owners (#67552 )

2021-10-29 12:42:01 -07:00

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

NNAPI: quant logistic fix (#70847 )

2022-01-07 13:36:33 -08:00

test_numba_integration.py

Remove unwanted comma (#71193 )

2022-01-13 19:09:05 -08:00

test_numpy_interop.py

ENH: Add a force argument to numpy() (#78564 )

2022-06-06 14:14:17 +00:00

test_openmp.py

Set test owners for tests with unknown owners (#67552 )

2021-10-29 12:42:01 -07: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_ops.py

Autogen Tags enum, and allow specifying tags while defining an op

2022-06-11 00:29: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

[skip ci] Set test owner for deploy/package tests (#66830 )

2021-10-26 07:49:33 -07:00

test_per_overload_api.py

Update __torch_dispatch__ to return op overload instead of the opoverload packet function (#72673 )

2022-03-07 22:38:42 +00:00

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

Revert "[PyTorch] Record Sequence Number to Match Forward and Backward Operators (#78795 )"

2022-06-10 13:11:44 +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

Set test owners for tests with unknown owners (#67552 )

2021-10-29 12:42:01 -07:00

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

Register torch.return_types.* as pytree nodes

2022-04-19 13:46:20 +00:00

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

Revert "Support multi-dimensional lengths in segment_reduce to support pytorch_scatter.segment_* functionalities (CUDA)"

2022-06-10 01:57:34 +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

Add ownership to more edge tests (#67859 )

2021-11-05 11:01:16 -07:00

test_shape_ops.py

[testing] Update dispatch macros (#74977 )

2022-03-30 14:13:21 -07:00

test_show_pickle.py

Add ownership to more edge tests (#67859 )

2021-11-05 11:01:16 -07:00

test_sort_and_select.py

MAINT: Harmonize argsort params with array_api (#75162 )

2022-06-09 12:32:01 +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_sparse.py

index_select for COO CUDA tensors. (#77551 )

2022-06-01 17:39:03 +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] Implement prim::Fork and aten::wait (#78780 )

2022-06-03 23:39:04 +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_pybind.py

[NNC] Lowering function generates the output buffer with the specified stride (#76529 )

2022-05-04 20:04:22 +00:00

test_tensorexpr.py

[NNC] channels last propagation within NNC fusion group (#76948 )

2022-05-30 18:31:49 +00:00

test_testing.py

Print output during MPS test import tests (#79163 )

2022-06-09 13:07:05 +00:00

test_throughput_benchmark.py

Set test owners for tests with unknown owners (#67552 )

2021-10-29 12:42:01 -07:00

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

[skip ci] Set test owners for test_typing.py (#66869 )

2021-10-20 09:41:13 -07:00

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

[skip ci] Set test owners for test_typing.py (#66869 )

2021-10-20 09:41:13 -07:00

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

update example in classification losses (#69816 )

2021-12-21 02:46:48 -08:00

test_vulkan.py

[skip ci] Set test owner for mobile tests (#66829 )

2021-10-26 10:20:01 -07:00

test_xnnpack_integration.py

Update XNNPACK (#73467 )

2022-02-28 06:09:26 +00:00