pytorch

ao/sparsity

[Quant] Add unified x86 quant backend (#84329 )

2022-09-29 00:44:40 +00:00

autograd

Fix autograd.functional tests to not fail with logging tensor

2022-04-20 20:32:40 +00:00

backends/xeon

A Launch script with Best Recipe of Deep Learning on Intel Xeon CPU (#63932 )

2022-07-29 12:57:22 +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

Implement col2im decomposition and fix im2col and add a few preconditions (#85541 )

2022-09-30 09:31:53 +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

Refactor distribuetd to use absolute header path (#85780 )

2022-09-30 05:13:50 +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

Add reduce_scatter_tensor in place of _reduce_scatter_base (#85867 )

2022-09-30 05:48:16 +00:00

distributions

Revert "Add mkl implementation for exponential on CPU (#69967 )"

2022-09-08 20:30:23 +00:00

error_messages

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

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

expect

[FX] Add one option to maintain the FX graph execution order after splitting_module (#85188 )

2022-09-23 23:21:54 +00:00

forward_backward_compatibility

Add python nested_tensor and as_nested_tensor constructors in torch.nested (#85593 )

2022-09-28 20:15:02 +00:00

fx

[quant][ao_migration] nn.intrinsic migration to ao (#84842 )

2022-09-28 23:54:29 +00:00

jit

[JIT] support freezing modules that don't have a forward method (#85779 )

2022-09-28 17:05:01 +00:00

jit_hooks

[ROCM] Enable custom tests on rocm

2022-04-26 14:22:14 +00:00

lazy

Revert "Revert "Symintifying slice ops (#85196 )"" (#85746 )

2022-09-28 04:37:35 +00:00

mobile

[quant][ao_migration] torch.nn.quantized.modules → torch.ao.nn.quantized.modules (#78713 )

2022-08-25 16:50:33 +00:00

nn

[fix] max_pool1d: shape check (#85594 )

2022-09-29 15:40:09 +00:00

onnx

[ONNX] Initial version of diagnostics infrastructure. (#85107 )

2022-09-30 07:47:26 +00:00

package

Allow torch._C to be recognized a module in torch.package (#80917 )

2022-07-12 18:10:22 +00:00

profiler

[Profiler] Extend ID assignment to allocations and frees (#85719 )

2022-09-30 04:39:08 +00:00

quantization

[qat]A more stable conv_bn fusion for qat training. (#85744 )

2022-09-29 07:58:54 +00:00

scripts

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

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

test_img

…

typing

Add typing for torch.return_type

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

allowlist_for_publicAPI.json

Pass group ranks and options to third party distributed backends (#73164 )

2022-09-29 17:28:58 +00:00

conftest.py

pytest to run test_ops, test_ops_gradients, test_ops_jit in non linux cuda environments (#79898 )

2022-07-19 19:50:57 +00:00

create_dummy_torchscript_model.py

Add test for FC/BC for torchscript file.

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

delete.py

Introducing DataChunk for DataPipes batching (#62768 )

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

functorch

Enable pytest-shard for functorch (#85321 )

2022-09-24 01:17:04 +00: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

mkl_verbose.py

[RFC] enable oneMKL&oneDNN on-demands verbose functinality (#63212 )

2022-07-27 23:29:35 +00:00

mkldnn_verbose.py

[RFC] enable oneMKL&oneDNN on-demands verbose functinality (#63212 )

2022-07-27 23:29:35 +00:00

run_doctests.sh

Integrate xdoctest - Rebased (#82797 )

2022-08-12 02:08:01 +00:00

run_test.py

Enable pytest-shard for functorch (#85321 )

2022-09-24 01:17:04 +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] Data Sparsity with Post Training Quantization (#82759 )

2022-08-10 16:51:35 +00:00

test_autocast.py

Update operator list for AutocastCPU (#68725 )

2022-05-11 17:28:35 +00:00

test_autograd.py

Add mechanism to disable the "saved tensors hooks" feature (#85553 )

2022-09-28 22:49:28 +00:00

test_binary_ufuncs.py

Add bfloat16 support for lerp on CPU (#84327 )

2022-09-29 01:16:16 +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_comparison_utils.py

Add aten _assert_tensor_metadata op (#84617 )

2022-09-19 20:48:09 +00:00

test_complex.py

[testing] Update dispatch macros (#74977 )

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

test_cpp_api_parity.py

Make M1 tests green (#82213 )

2022-08-05 16:12:08 +00: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

Reenable TestCppExtensionJIT on M1 (#84552 )

2022-09-06 17:49:29 +00:00

test_cpp_extensions_open_device_registration.py

Make M1 tests green (#82213 )

2022-08-05 16:12:08 +00:00

test_cuda_primary_ctx.py

remove hasPrimaryContext workaround on ROCm (#71146 )

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

test_cuda_sanitizer.py

Rework printing tensor aliases in CSAN error message (#85008 )

2022-09-21 13:41:52 +00:00

test_cuda_trace.py

Add synchronize hooks (#84427 )

2022-09-09 13:56:59 +00:00

test_cuda.py

resubmit: "resubmit: [mta] APEX style Fused Adam (#81705 ) (#85507 )" (#85739 )

2022-09-29 16:58:59 +00:00

test_dataloader.py

[DataPipe] Reset Shuffler's iterator when NotStarted (#83535 )

2022-08-25 19:45:41 +00:00

test_datapipe.py

[DataPipe] Fix MapDataPipe spawn lambda test (#85668 )

2022-09-27 19:58:15 +00:00

test_decomp.py

Revert "Improve make_tensor performance for float and complex types (#85473 )"

2022-09-29 20:06: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

Remove unnecessary skips in test_dispatch.py (#85557 )

2022-09-26 15:35:47 +00:00

test_dlpack.py

Normalize DLPack stride to 1 where shape < 2 (#83158 )

2022-08-23 15:03:29 +00:00

test_dynamic_shapes.py

Slightly beefed up dynamic shapes tests for storage_offset (#85806 )

2022-09-28 19:25:22 +00:00

test_expanded_weights.py

[Expanded Weights] add 'same' and 'valid' padding support (#83345 )

2022-08-16 22:39:08 +00:00

test_fake_tensor.py

Handle fake tensor in intlist (#85759 )

2022-09-28 21:58:54 +00:00

test_foreach.py

Test Dynamo CI w Fake Tensors (#84282 )

2022-09-01 00:15:05 +00:00

test_function_schema.py

[frontend] Print real type for Argument (#85103 )

2022-09-15 22:16:32 +00:00

test_functional_autograd_benchmark.py

fix python version error (#71021 )

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

test_functional_optim.py

Enable Zero1's ddp_with_overlap for hpu backend (#80438 )

2022-07-18 15:05:27 +00:00

test_functionalization.py

Refactored proxytensor to clean up separate branches (#84325 )

2022-08-31 09:37:55 +00:00

test_futures.py

[torch.futures] Fix nullptr deref (#85304 )

2022-09-20 01:49:04 +00:00

test_fx_backends.py

[ROCm] enable nvfuser (#82498 )

2022-08-30 21:50:39 +00:00

test_fx_experimental.py

[fx] Make NormalizeArgs preserve node type (#85637 )

2022-09-26 21:30:16 +00:00

test_fx_passes.py

Support matching Args for SubgraphMatcher (#85456 )

2022-09-24 20:06:48 +00:00

test_fx_reinplace_pass.py

[CMake] Add functorch target (#83464 )

2022-09-14 00:05:33 +00:00

test_fx.py

Implement col2im decomposition and fix im2col and add a few preconditions (#85541 )

2022-09-30 09:31:53 +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

Revert "Revert "[ci] remove remaining RDS dependency""

2022-06-15 10:04:13 -07:00

test_indexing.py

Re-enable fixed dynamo tests (#84969 )

2022-09-16 15:36:52 +00:00

test_itt.py

add itt unit test and docstrings (#84848 )

2022-09-28 01:39:58 +00:00

test_jit_autocast.py

[JIT] Retry - Support scripting torch.is_autocast_enabled() (#82394 )

2022-08-10 18:26:17 +00:00

test_jit_cuda_fuser.py

[NVFuser] Upstream push 0907 (#84626 )

2022-09-23 20:29:48 +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

Optimize to if the datatyep of the source tensor is as same as the dest datatype (#85140 )

2022-09-27 04:40:42 +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

add oneDNN graph fuser context API and unittest (#82491 )

2022-09-12 20:09:00 +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

Update Dynamo pin (#83829 )

2022-08-26 20:49:43 +00:00

test_jiterator.py

[ROCm] enable jiterator (#77982 )

2022-08-15 16:04:09 +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

Add path optimize kwarg to einsum (#84890 )

2022-09-24 03:47:36 +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

[maskedtensor] port torch/_masked into torch/masked (#85515 )

2022-09-26 23:41:13 +00:00

test_maskedtensor.py

[maskedtensor] port torch/_masked into torch/masked (#85515 )

2022-09-26 23:41:13 +00:00

test_meta.py

Implement col2im decomposition and fix im2col and add a few preconditions (#85541 )

2022-09-30 09:31:53 +00:00

test_metal.py

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

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

test_mkl_verbose.py

[RFC] enable oneMKL&oneDNN on-demands verbose functinality (#63212 )

2022-07-27 23:29:35 +00:00

test_mkldnn_fusion.py

[NNC] support aten::_convolution when it is 2D conv (#84038 )

2022-09-19 17:45:20 +00:00

test_mkldnn_verbose.py

[RFC] enable oneMKL&oneDNN on-demands verbose functinality (#63212 )

2022-07-27 23:29:35 +00:00

test_mkldnn.py

test mkldnn conv2d channels last when weight is nchw format (#77348 )

2022-09-09 07:28:04 +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

Fix failing test_model_dump due to empty file (#84744 )

2022-09-12 20:59:19 +00:00

test_module_init.py

[quant][ao_migration] torch.nn.qat → torch.ao.nn.qat (#78716 )

2022-08-25 16:50:38 +00:00

test_modules.py

TestModule: Don't assume sample inputs version counter is zero (#85734 )

2022-09-28 19:32:06 +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] Handle scalar input for scatter and gather (#85842 )

2022-09-30 00:24:16 +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

[ROCm] Enable test_multiprocessing tests (#82356 )

2022-08-24 20:49:20 +00:00

test_namedtensor.py

Revert "Update amax/amin/norm/count_nonzero signatures with int[*]? dim (#83300 )"

2022-09-28 17:04:53 +00:00

test_namedtuple_return_api.py

Remove deprecated torch.lstsq (#70980 )

2022-09-23 00:16:55 +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

Add python nested_tensor and as_nested_tensor constructors in torch.nested (#85593 )

2022-09-28 20:15:02 +00:00

test_nestedtensor.py

Allow only one -1 in nested view/reshape (#85691 )

2022-09-28 22:29:40 +00:00

test_nn.py

Implement col2im decomposition and fix im2col and add a few preconditions (#85541 )

2022-09-30 09:31:53 +00:00

test_nnapi.py

[quant][ao_migration] torch.nn.quantized.modules → torch.ao.nn.quantized.modules (#78713 )

2022-08-25 16:50:33 +00: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_nvfuser_frontend.py

Create Cache for Fusion Reuse in NVFuser in Python Frontend for Primtorch (#85045 )

2022-09-17 10:52:54 +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

Reenable slow gradcheck and make it pass (#80514 )

2022-07-22 02:05:37 +00:00

test_ops_jit.py

OpInfo for Slice (#85554 )

2022-09-23 22:01:32 +00:00

test_ops.py

Turn on aliasing tests for fake backwards, Fix Batch norm running mean/var decomp aliasing (#85471 )

2022-09-28 23:06:59 +00:00

test_optim.py

resubmit: "resubmit: [mta] APEX style Fused Adam (#81705 ) (#85507 )" (#85739 )

2022-09-29 16:58:59 +00:00

test_overrides.py

[Modes] fix handle_torch_funcion logic (#85707 )

2022-09-27 18:35:51 +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

Add OpOverload.decompose API (#83075 )

2022-08-09 18:53:19 +00:00

test_prims.py

Add a way to skip lowering to nvprims (#85811 )

2022-09-30 12:01:45 +00:00

test_proxy_tensor.py

Implement duck shaping on SymInts (#85808 )

2022-09-29 00:45: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

More SymFloat support (#85411 )

2022-09-22 08:07:22 +00:00

test_python_dispatch.py

[Modes] remove enable and rewrite mode stack (squashed) (#84774 )

2022-09-27 01:04:35 +00:00

test_pytree.py

Add *_only and all/any pytree utilities (#83316 )

2022-08-12 17:31:55 +00:00

test_quantization.py

[quant][ao_migration] nn.intrinsic migration to ao (#84842 )

2022-09-28 23:54:29 +00:00

test_reductions.py

assert weights being 1-d tensor in bincount (#85881 )

2022-09-29 21:20:38 +00:00

test_scatter_gather_ops.py

Add deterministic impl of scatter_add CUDA for all input sizes (#79466 )

2022-09-07 03:12:49 +00:00

test_schema_check.py

[Modes] remove enable and rewrite mode stack (squashed) (#84774 )

2022-09-27 01:04:35 +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

Rename _Typed/_UntypedStorage to Typed/UntypedStorage and update docs (#82438 )

2022-07-30 19:37:08 +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

Setup for running PyTorch tests with TorchDynamo and skips for known failing tests (#80106 )

2022-07-07 18:57:33 +00: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

Re-enable fixed dynamo tests (#84969 )

2022-09-16 15:36:52 +00:00

test_sparse_csr.py

(bsr/csr) x dense mm (#85551 )

2022-09-29 17:12:04 +00:00

test_sparse.py

[maskedtensor] port torch/_masked into torch/masked (#85515 )

2022-09-26 23:41:13 +00:00

test_spectral_ops.py

Fix use-dict-literal lint (#83718 )

2022-08-24 00:26:46 +00:00

test_stateless.py

Make nn.stateless correctly reset parameters if the forward pass fails (#81262 )

2022-07-12 13:54:23 +00:00

test_static_runtime.py

[Static Runtime] test case for staticRuntime::runAsync() API (#80407 )

2022-07-05 23:40:53 +00:00

test_subclass.py

[torchdynamo hash update] update the pinned torchdynamo hash (#85225 )

2022-09-26 20:07:13 +00:00

test_tensor_creation_ops.py

[reland] [numpy] add torch.concatenate, alias of torch.cat (#85073 )

2022-09-15 19:34:44 +00:00

test_tensorboard.py

[tensorboard] update assertion error for scalar() and fix docs (#76859 )

2022-06-16 05:24:20 +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

Use high precision accmulate buffer for bf16 accmulation (#84402 )

2022-09-27 04:40:42 +00:00

test_testing.py

Move OpInfo class into new opinfo folder (#82540 )

2022-08-05 15:10:17 +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

Add python nested_tensor and as_nested_tensor constructors in torch.nested (#85593 )

2022-09-28 20:15:02 +00:00

test_transformers.py

Add python nested_tensor and as_nested_tensor constructors in torch.nested (#85593 )

2022-09-28 20:15:02 +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

Land "Make ceil,floor,round,trunc handle integers" (#85144 )

2022-09-21 17:23:47 +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

Adding multigammaln ref and fix arange (#85153 )

2022-09-20 17:52:56 +00:00

test_utils.py

Fix inheritance in TestDataLoaderUtil (#85018 )

2022-09-14 22:04:43 +00:00

test_view_ops.py

Setup for running PyTorch tests with TorchDynamo and skips for known failing tests (#80106 )

2022-07-07 18:57:33 +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