| .. |
|
ao/sparsity
|
[ao][sparsity] comsability for sparsity and QAT convert
|
2022-04-11 16:32:08 +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
|
|
|
|
cpp
|
[Bootcamp]Add option for flatbuffer loader to copy memory to individual tensors (#76986)
|
2022-05-09 17:29:30 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
[Reland] Add BUILD_LAZY_CUDA_LINALG option (#73447)
|
2022-03-01 21:33:07 +00:00 |
|
custom_backend
|
[ROCM] Enable custom tests on rocm
|
2022-04-26 14:22:14 +00:00 |
|
custom_operator
|
[ROCM] Enable custom tests on rocm
|
2022-04-26 14:22:14 +00:00 |
|
distributed
|
Additional ops for ShardedTensor, ReplicatedTensor and PartialTensor.
|
2022-05-06 16:28:04 +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
|
Revert "stft: remove non-center overload and python functional wrapper"
|
2022-05-09 19:59:46 +00:00 |
|
fx
|
[test_fx_const_fold] Remove dependencies on acc_* (#72810)
|
2022-02-14 20:46:54 +00:00 |
|
jit
|
Revert "stft: remove non-center overload and python functional wrapper"
|
2022-05-09 19:59:46 +00:00 |
|
jit_hooks
|
[ROCM] Enable custom tests on rocm
|
2022-04-26 14:22:14 +00:00 |
|
lazy
|
Dynamo+LTC: handle inplace ops (#75359)
|
2022-04-08 03:14:28 +00:00 |
|
mobile
|
Revert "Deprecate torch.lu"
|
2022-05-09 19:06:44 +00:00 |
|
onnx
|
[ONNX] Format ONNX python with black
|
2022-05-05 00:19:22 +00:00 |
|
package
|
Revert D35423079: [pkg] add generic ZipFile Reader/Writer
|
2022-04-07 13:33:47 +00:00 |
|
quantization
|
ns for fx: skip shadowing ops if copy subgraph is not implemented (#76663)
|
2022-05-05 13:19:53 +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 "Add linalg.lu_solve"
|
2022-05-09 19:12:30 +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
|
Additional ops for ShardedTensor, ReplicatedTensor and PartialTensor.
|
2022-05-06 16:28:04 +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
|
Bugfix scatter_reduce backward formulas
|
2022-05-05 20:22:39 +00:00 |
|
test_binary_ufuncs.py
|
Adds decorators for Python References and extends Python Reference testing (#76945)
|
2022-05-07 03:42:24 +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
|
Remove breakpad dependency
|
2022-05-03 20:21:55 +00:00 |
|
test_cuda.py
|
Add high level control of fp32 matmul precision; disable TF32 for matmuls by default
|
2022-05-04 20:40:13 +00:00 |
|
test_cuda_primary_ctx.py
|
remove hasPrimaryContext workaround on ROCm (#71146)
|
2022-01-25 20:32:12 +00:00 |
|
test_dataloader.py
|
Update Dataloader with default parameter device (#65402)
|
2022-04-21 01:33:53 +00:00 |
|
test_datapipe.py
|
Revert [DataPipe] Update mux data pipe
|
2022-04-28 00:06:30 +00:00 |
|
test_decomp.py
|
Register PrimTorch refs as decompositions.
|
2022-05-06 20:11:45 +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: instance norm faster rule
|
2022-04-19 19:40:09 +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
|
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
|
Register PrimTorch refs as decompositions.
|
2022-05-06 20:11:45 +00:00 |
|
test_futures.py
|
|
|
|
test_fx.py
|
Back out "record_function: update to use custom_class API" (#76253)
|
2022-04-26 17:49:57 -04:00 |
|
test_fx_experimental.py
|
Added proxy tensor
|
2022-05-03 22:46:30 +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
|
Fix deterministic indexing with non-contiguous tensor
|
2022-04-22 18:46:50 +00:00 |
|
test_jit.py
|
Revert "Deprecate torch.lu"
|
2022-05-09 19:06:44 +00:00 |
|
test_jit_autocast.py
|
[JIT][Autocast] Don't cast softmax on CPU
|
2022-05-02 22:47:52 +00:00 |
|
test_jit_cuda_fuser.py
|
Torchvision patch (#77001)
|
2022-05-09 16:53:23 +00:00 |
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
[primTorch] flatten, squeeze, unsqueeze... (#77043)
|
2022-05-09 11:24:55 +00:00 |
|
test_jit_legacy.py
|
|
|
|
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
|
|
|
|
test_jit_simple.py
|
|
|
|
test_jit_string.py
|
|
|
|
test_jiterator.py
|
Python Interface for Jiterator
|
2022-05-06 18:44:28 +00:00 |
|
test_kernel_launch_checks.py
|
|
|
|
test_license.py
|
|
|
|
test_linalg.py
|
Revert "Add linalg.lu_solve"
|
2022-05-09 19:12: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
|
Add OpInfo based meta tensor tests [RELAND]
|
2022-05-07 12:15:10 +00:00 |
|
test_metal.py
|
|
|
|
test_mkldnn.py
|
Revert D29463782: opitimze ConvTransposedND with mkldnn float32 and bfloat16 on CPU
|
2022-05-06 19:34:41 +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
|
fx quant: add quantized Softmax workflow integration (#75106)
|
2022-04-20 21:54:26 +00:00 |
|
test_modules.py
|
only compare attributes for meta tensors (#72508)
|
2022-02-17 02:33:08 +00:00 |
|
test_monitor.py
|
torch/monitor: merge Interval and FixedCount stats (#72009)
|
2022-01-30 23:21:59 +00:00 |
|
test_multiprocessing.py
|
[ROCm] Skipping few multiprocess test
|
2022-04-27 17:56:36 +00:00 |
|
test_multiprocessing_spawn.py
|
Make test_multiprocessing_spawn.py compatible with pytest (#50408)
|
2022-02-10 16:18:46 +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
|
Add linalg.lu
|
2022-05-05 09:17:05 +00:00 |
|
test_native_functions.py
|
|
|
|
test_native_mha.py
|
[PyTorch] Add native fast path for transformer encoder inference (#76333)
|
2022-04-26 12:58:03 -04:00 |
|
test_nestedtensor.py
|
Adding output_size to to_padded_tensor (#76640)
|
2022-05-03 18:22:51 +00:00 |
|
test_nn.py
|
[PyTorch] Don't enter MHA fast path when bias & query dtypes don't match
|
2022-05-09 18:21:04 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
Remove unwanted comma (#71193)
|
2022-01-13 19:09:05 -08:00 |
|
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
|
[primTorch] flatten, squeeze, unsqueeze... (#77043)
|
2022-05-09 11:24:55 +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
|
Fix CosineAnnealingLR on restart
|
2022-04-20 13:35:01 +00:00 |
|
test_overrides.py
|
Cannibalize noarch CI job into crossref CI job
|
2022-04-20 11:56:25 +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_prims.py
|
Adds python ref consistency test, elementwise unary reference inputs, and formats test files
|
2022-05-01 22:42:46 +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
|
Add "mps" device to PyTorch framework.
|
2022-04-27 19:21:57 +00:00 |
|
test_python_dispatch.py
|
Register PrimTorch refs as decompositions.
|
2022-05-06 20:11:45 +00:00 |
|
test_pytree.py
|
Register torch.return_types.* as pytree nodes
|
2022-04-19 13:46:20 +00:00 |
|
test_quantization.py
|
[quant][fx] Add RNN reference module (#73386)
|
2022-03-02 10:30:37 +00:00 |
|
test_reductions.py
|
Various OpInfo architecture improvements
|
2022-04-18 21:55:32 +00:00 |
|
test_scatter_gather_ops.py
|
Bugfix NAN and Inf handling for scatter_reduce (amin and amax)
|
2022-05-06 15:39:57 +00:00 |
|
test_segment_reductions.py
|
|
|
|
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
|
[testing] Update dispatch macros (#74977)
|
2022-03-30 14:13:21 -07:00 |
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
Port sort to structured kernels.
|
2022-04-21 01:48:46 +00:00 |
|
test_sparse.py
|
Faster index_select for sparse COO tensors on CPU. (#72710)
|
2022-05-09 19:59:39 +00:00 |
|
test_sparse_csr.py
|
Sparse CSR: Fix sampled_addmm for noncontiguous inputs and fix block sparse triangular solve
|
2022-05-05 09:00:48 +00:00 |
|
test_spectral_ops.py
|
[ROCM] remove rocfft workaround
|
2022-05-04 19:43:46 +00:00 |
|
test_stateless.py
|
Removed direct doc formatting
|
2022-05-02 14:14:33 +00:00 |
|
test_static_runtime.py
|
Fix quicklints (#72256)
|
2022-02-03 15:17:35 +00:00 |
|
test_subclass.py
|
Support for tensor subclasses as parameters
|
2022-04-27 19:28:55 +00:00 |
|
test_tensor_creation_ops.py
|
Adds python ref consistency test, elementwise unary reference inputs, and formats test files
|
2022-05-01 22:42:46 +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] call super().setUp() & tearDown() in test_tensorexpr.py (#74504)
|
2022-03-22 20:17:21 +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
|
add rocm ciflow/slow workflow
|
2022-04-22 17:41:28 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
Support torch.Tensor.to for CSR
|
2022-05-05 21:59:50 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
|
|
|
test_type_promotion.py
|
type promote clamp (#77035)
|
2022-05-09 05:54:17 +00:00 |
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
Adds python ref consistency test, elementwise unary reference inputs, and formats test files
|
2022-05-01 22:42:46 +00:00 |
|
test_utils.py
|
Fix issue with _checkpoint_without_reentrant
|
2022-05-05 17:37:31 +00:00 |
|
test_view_ops.py
|
Adds python ref consistency test, elementwise unary reference inputs, and formats test files
|
2022-05-01 22:42:46 +00:00 |
|
test_vmap.py
|
|
|
|
test_vulkan.py
|
|
|
|
test_xnnpack_integration.py
|
Update XNNPACK (#73467)
|
2022-02-28 06:09:26 +00:00 |
2c5bf12584