| .. |
|
ao/sparsity
|
[ao][sparsity] comsability for sparsity and QAT convert
|
2022-04-11 16:32:08 +00:00 |
|
autograd
|
Miscellaneous __torch_function__ fixes
|
2022-04-11 16:52:16 +00:00 |
|
benchmark_utils
|
Add CPU slow test job (#73748)
|
2022-03-23 21:17:27 +00:00 |
|
bottleneck_test
|
|
|
|
cpp
|
Reland "Make debug_pkl smaller by only emitting unique traces." (#73368)
|
2022-04-18 22:34:21 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
|
|
|
custom_backend
|
|
|
|
custom_operator
|
|
|
|
distributed
|
[FSDP] Fix no_sync() + FULL_SHARD root all-gather behavior
|
2022-04-18 20:49:50 +00:00 |
|
distributions
|
Improve numerical stability of torch.distributions.wishart.Wishart (#72993)
|
2022-03-15 18:30:08 +00:00 |
|
error_messages
|
|
|
|
expect
|
Fix ONNX ATen fallback for non-caffe2 engines
|
2022-04-14 23:18:45 +00:00 |
|
forward_backward_compatibility
|
Port cat kernel to structured kernels.
|
2022-04-14 17:49:43 +00:00 |
|
fx
|
|
|
|
jit
|
Add upgrader related logic to flatbuffer (#71451)
|
2022-04-17 18:51:23 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
Dynamo+LTC: handle inplace ops (#75359)
|
2022-04-08 03:14:28 +00:00 |
|
mobile
|
[mobile] Update test model generation script to count op occurrences
|
2022-04-08 05:54:37 +00:00 |
|
onnx
|
Fix ONNX ATen fallback for non-caffe2 engines
|
2022-04-14 23:18:45 +00:00 |
|
package
|
Revert D35423079: [pkg] add generic ZipFile Reader/Writer
|
2022-04-07 13:33:47 +00:00 |
|
quantization
|
[quant][core][gpu][bug-fix] Added additional caching support in quantized cudnn add_relu op
|
2022-04-18 16:53:18 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
typing
|
Add typing for torch.return_type
|
2022-03-29 02:17:21 +00:00 |
|
allowlist_for_publicAPI.json
|
Revert "Reland Fix public binding check for modules with __all__"
|
2022-04-19 15:36:29 +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
|
|
|
|
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
|
[shard] Add basic math ops to ShardedTensor and add ReplicatedTensor inter-op
|
2022-04-12 04:25:10 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
[ao][sparsity] make sparsity and PTQ compose (#74845)
|
2022-04-05 03:35:41 +00:00 |
|
test_autocast.py
|
|
|
|
test_autograd.py
|
Fix complex to real casting warning in _to_copy backward
|
2022-04-19 14:04:13 +00:00 |
|
test_binary_ufuncs.py
|
Various OpInfo architecture improvements
|
2022-04-18 21:55:32 +00:00 |
|
test_bundled_images.py
|
|
|
|
test_bundled_inputs.py
|
|
|
|
test_complex.py
|
[testing] Update dispatch macros (#74977)
|
2022-03-30 14:13:21 -07:00 |
|
test_cpp_api_parity.py
|
|
|
|
test_cpp_extensions_aot.py
|
|
|
|
test_cpp_extensions_jit.py
|
Revert "Remove breakpad dependency"
|
2022-04-17 17:58:51 +00:00 |
|
test_cuda.py
|
Fix test_reduce_add_coalesced failure (#74027)
|
2022-03-15 06:29:54 +00:00 |
|
test_cuda_primary_ctx.py
|
|
|
|
test_dataloader.py
|
don't add extra shuffle in DataLoader2 if one is present
|
2022-04-05 19:53:08 +00:00 |
|
test_datapipe.py
|
[DataPipe] apply dill serialization for _Demux and add cache to traverse
|
2022-04-04 19:45:14 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
Towards supporting quantized structured kernels (#74560)
|
2022-04-05 04:31:48 +00:00 |
|
test_expanded_weights.py
|
expanded weights: group norm faster rule
|
2022-03-31 20:06:54 +00:00 |
|
test_foreach.py
|
[testing] Update dispatch macros (#74977)
|
2022-03-30 14:13:21 -07:00 |
|
test_function_schema.py
|
|
|
|
test_functional_autograd_benchmark.py
|
|
|
|
test_functional_optim.py
|
|
|
|
test_functionalization.py
|
avoid some unnecessary view_copy calls
|
2022-04-18 20:38:55 +00:00 |
|
test_futures.py
|
|
|
|
test_fx.py
|
record_function: update to use custom_class API
|
2022-03-30 15:57:28 +00:00 |
|
test_fx_experimental.py
|
Various OpInfo architecture improvements
|
2022-04-18 21:55:32 +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
|
|
|
|
test_indexing.py
|
[Codemod][Codemod deprecated unittest asserts] fbcode//caffe2/test (#71708)
|
2022-03-15 19:28:52 +00:00 |
|
test_jit.py
|
Various OpInfo architecture improvements
|
2022-04-18 21:55:32 +00:00 |
|
test_jit_autocast.py
|
[JIT] Add autocasting to freezing pass & enable autocast pass by default (#74178)
|
2022-03-23 23:10:48 +00:00 |
|
test_jit_cuda_fuser.py
|
Enable windows tests for nvfuser
|
2022-04-19 12:36:50 +00:00 |
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
Revert D34808051: [tensorexpr] Enabled aten::stack in the fuser pass with static shapes
|
2022-03-31 04:25:43 +00:00 |
|
test_jit_legacy.py
|
|
|
|
test_jit_profiling.py
|
|
|
|
test_jit_simple.py
|
|
|
|
test_jit_string.py
|
|
|
|
test_kernel_launch_checks.py
|
|
|
|
test_license.py
|
|
|
|
test_linalg.py
|
remove fp16 support from cpu linalg functions
|
2022-04-18 15:55:38 +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
|
Support masked sum on CSR tensors [CPU, CUDA]
|
2022-04-08 20:07:18 +00:00 |
|
test_metal.py
|
|
|
|
test_mkldnn.py
|
|
|
|
test_mobile_optimizer.py
|
|
|
|
test_model_dump.py
|
|
|
|
test_module_init.py
|
[quant][fx] Support conv1d and its fusion variants in QAT (#74506)
|
2022-03-23 18:43:53 +00:00 |
|
test_modules.py
|
|
|
|
test_monitor.py
|
|
|
|
test_multiprocessing.py
|
[ROCm] enable HIP IPC
|
2022-03-21 19:32:01 +00:00 |
|
test_multiprocessing_spawn.py
|
|
|
|
test_namedtensor.py
|
|
|
|
test_namedtuple_return_api.py
|
Add C++ implementation of histogramdd
|
2022-03-29 02:17:21 +00:00 |
|
test_native_functions.py
|
|
|
|
test_nestedtensor.py
|
[PyTorch] Add NestedTensor support functions for transformers
|
2022-04-14 16:30:23 +00:00 |
|
test_nn.py
|
add operator header
|
2022-04-19 15:23:25 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
torch.tensor: add tests for list of numpy arrays case
|
2022-04-04 01:02:31 +00:00 |
|
test_openmp.py
|
|
|
|
test_ops.py
|
[ROCm] enable composite compliance backward tests
|
2022-04-19 13:47:06 +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
|
Add maximize flag to Adadelta
|
2022-04-08 20:32:35 +00:00 |
|
test_overrides.py
|
Remove histogramdd functional wrapper
|
2022-04-14 20:56:24 +00:00 |
|
test_package.py
|
|
|
|
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_profiler.py
|
[Profiler] Expose profilerType in Python
|
2022-04-16 21:08:18 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
Revert "Reland Fix public binding check for modules with __all__"
|
2022-04-19 15:36:29 +00:00 |
|
test_python_dispatch.py
|
Preserve Python dispatch keys upon copy_tensor_metadata_except_version_counter
|
2022-04-15 13:26:23 +00:00 |
|
test_pytree.py
|
Register torch.return_types.* as pytree nodes
|
2022-04-19 13:46:20 +00:00 |
|
test_quantization.py
|
|
|
|
test_reductions.py
|
Various OpInfo architecture improvements
|
2022-04-18 21:55:32 +00:00 |
|
test_scatter_gather_ops.py
|
Add include_self flag to scatter_reduce
|
2022-04-05 16:31:39 +00:00 |
|
test_segment_reductions.py
|
|
|
|
test_serialization.py
|
[testing] Update dispatch macros (#74977)
|
2022-03-30 14:13:21 -07:00 |
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
[testing] Update dispatch macros (#74977)
|
2022-03-30 14:13:21 -07:00 |
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
[testing] Update dispatch macros (#74977)
|
2022-03-30 14:13:21 -07:00 |
|
test_sparse.py
|
Enable test_sparse_mask for Windows
|
2022-04-11 17:21:29 +00:00 |
|
test_sparse_csr.py
|
Validate Sparse Compressed tensor arguments
|
2022-04-18 02:21:22 +00:00 |
|
test_spectral_ops.py
|
Various OpInfo architecture improvements
|
2022-04-18 21:55:32 +00:00 |
|
test_stateless.py
|
|
|
|
test_static_runtime.py
|
|
|
|
test_tensor_creation_ops.py
|
Port cat kernel to structured kernels.
|
2022-04-14 17:49:43 +00:00 |
|
test_tensorboard.py
|
[Codemod][Codemod deprecated unittest asserts] fbcode//caffe2/test (#71708)
|
2022-03-15 19:28:52 +00:00 |
|
test_tensorexpr.py
|
[NNC] call super().setUp() & tearDown() in test_tensorexpr.py (#74504)
|
2022-03-22 20:17:21 +00:00 |
|
test_tensorexpr_pybind.py
|
Revert D34808051: [tensorexpr] Enabled aten::stack in the fuser pass with static shapes
|
2022-03-31 04:25:43 +00:00 |
|
test_testing.py
|
Various OpInfo architecture improvements
|
2022-04-18 21:55:32 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
Remove noarchTest decorator
|
2022-04-19 00:48:49 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
|
|
|
test_type_promotion.py
|
[testing] Update dispatch macros (#74977)
|
2022-03-30 14:13:21 -07:00 |
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
Various OpInfo architecture improvements
|
2022-04-18 21:55:32 +00:00 |
|
test_utils.py
|
Revert "Remove breakpad dependency"
|
2022-04-17 17:58:51 +00:00 |
|
test_view_ops.py
|
code-generate non-aliasing {view}_copy kernels (#73442)
|
2022-04-11 19:48:55 +00:00 |
|
test_vmap.py
|
|
|
|
test_vulkan.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
