| .. |
|
ao/sparsity
|
[ao][sparsity] Refactor and bugfix SparseQuantizedLayer tests (#81802)
|
2022-08-03 01:12:32 +00:00 |
|
autograd
|
|
|
|
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
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
Use flatbuffer of alternate namespace (#82952)
|
2022-08-09 07:40:59 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
add open device registration test with cpp extensions (#80477)
|
2022-07-12 01:46:16 +00:00 |
|
custom_backend
|
|
|
|
custom_operator
|
|
|
|
distributed
|
Fix FSDP device_id when CPU offloading (#82892)
|
2022-08-07 19:06:32 +00:00 |
|
distributions
|
Ensure Transform is pickleable. (#81707)
|
2022-07-22 06:33:53 +00:00 |
|
error_messages
|
|
|
|
expect
|
reland "add a reinplacing FX pass (#80897)" (#82407)
|
2022-08-02 18:03:29 +00:00 |
|
forward_backward_compatibility
|
Add torch.unflatten and improve its docs (#81399)
|
2022-07-29 15:02:42 +00:00 |
|
fx
|
add the operations needed for electra model (#82856)
|
2022-08-08 16:56:06 +00:00 |
|
jit
|
Get rid of ENABLE_UPGRADERS macro (#77574)
|
2022-08-09 05:33:14 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
fix LTC usage of torch.tensor ctr, add test (#81928)
|
2022-07-22 13:57:53 +00:00 |
|
mobile
|
Update model_ops.yaml (#82444)
|
2022-08-02 20:04:49 +00:00 |
|
onnx
|
[ONNX] Reland #81953 Type utility for converting among JIT, torch and ONNX data types (#82995)
|
2022-08-08 23:43:43 +00:00 |
|
package
|
Allow torch._C to be recognized a module in torch.package (#80917)
|
2022-07-12 18:10:22 +00:00 |
|
quantization
|
[Quant][fx][bc-breaking] Integrate BackendConfig with quantization flow (part 2) (#82557)
|
2022-08-08 18:55:50 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
typing
|
|
|
|
allowlist_for_publicAPI.json
|
Rename _Typed/_UntypedStorage to Typed/UntypedStorage and update docs (#82438)
|
2022-07-30 19:37:08 +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
|
|
|
|
defs.bzl
|
|
|
|
delete.py
|
|
|
|
HowToWriteTestsUsingFileCheck.md
|
|
|
|
linear.py
|
|
|
|
load_torchscript_model.py
|
|
|
|
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 |
|
profiler_utils_mock_events.json
|
[Profiler] Add heuristic to rank events base on computed metrics (#80094)
|
2022-07-05 22:06:02 +00:00 |
|
run_test.py
|
Do not assume what is in os.environ (#82495)
|
2022-07-29 22:57:18 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
Make M1 tests green (#82213)
|
2022-08-05 16:12:08 +00:00 |
|
test_autocast.py
|
|
|
|
test_autograd.py
|
[ROCm] work-around missing hipProfilerStart/Stop (#82778)
|
2022-08-08 18:25:13 +00:00 |
|
test_binary_ufuncs.py
|
Allow ufunc OpInfos to have no reference (#82348)
|
2022-08-09 04:38:17 +00:00 |
|
test_bundled_images.py
|
|
|
|
test_bundled_inputs.py
|
|
|
|
test_complex.py
|
|
|
|
test_cpp_api_parity.py
|
Make M1 tests green (#82213)
|
2022-08-05 16:12:08 +00:00 |
|
test_cpp_extensions_aot.py
|
|
|
|
test_cpp_extensions_jit.py
|
Make M1 tests green (#82213)
|
2022-08-05 16:12:08 +00:00 |
|
test_cpp_extensions_open_device_registration.py
|
Make M1 tests green (#82213)
|
2022-08-05 16:12:08 +00:00 |
|
test_cuda.py
|
[Re-land] [CUDA graphs] Clear autocast amp cache (#81896)
|
2022-08-02 23:22:00 +00:00 |
|
test_cuda_primary_ctx.py
|
|
|
|
test_dataloader.py
|
Make M1 tests green (#82213)
|
2022-08-05 16:12:08 +00:00 |
|
test_datapipe.py
|
[DataPipe] Correct the type of exception that is being raised by ShufflerMapDataPipe (#82666)
|
2022-08-03 19:05:17 +00:00 |
|
test_decomp.py
|
Add ref for meshgrid (#82284)
|
2022-08-04 01:40:44 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
Make M1 tests green (#82213)
|
2022-08-05 16:12:08 +00:00 |
|
test_dynamic_shapes.py
|
add integer divison for symints (#82791)
|
2022-08-04 20:00:51 +00:00 |
|
test_dynamo_cudagraphs.py
|
Copy black config to ufmt and run lintrunner -a (#82043)
|
2022-07-23 06:04:21 +00:00 |
|
test_expanded_weights.py
|
[Expanded Weights] add ability to not specify batch size (#80944)
|
2022-07-19 13:43:46 +00:00 |
|
test_fake_tensor.py
|
Update nan_to_num prims (#82908)
|
2022-08-08 14:28:28 +00:00 |
|
test_foreach.py
|
[foreach][mta] Inplace maximum and minimum (#82523)
|
2022-08-03 03:40:42 +00:00 |
|
test_function_schema.py
|
|
|
|
test_functional_autograd_benchmark.py
|
|
|
|
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
|
Remove manual bindings for arange (#81380)
|
2022-08-07 00:10:27 +00:00 |
|
test_futures.py
|
|
|
|
test_fx.py
|
Preseve stack trace in nodes during fx.Transform (#82670)
|
2022-08-03 20:24:19 +00:00 |
|
test_fx_backends.py
|
Follow GitHub relabeling of oncall: fx for test owners (#81821)
|
2022-07-21 01:50:06 +00:00 |
|
test_fx_experimental.py
|
Follow GitHub relabeling of oncall: fx for test owners (#81821)
|
2022-07-21 01:50:06 +00:00 |
|
test_fx_passes.py
|
Follow GitHub relabeling of oncall: fx for test owners (#81821)
|
2022-07-21 01:50:06 +00:00 |
|
test_fx_reinplace_pass.py
|
reland "add a reinplacing FX pass (#80897)" (#82407)
|
2022-08-02 18:03:29 +00:00 |
|
test_hub.py
|
|
|
|
test_import_stats.py
|
|
|
|
test_indexing.py
|
|
|
|
test_jit.py
|
Get rid of ENABLE_UPGRADERS macro (#77574)
|
2022-08-09 05:33:14 +00:00 |
|
test_jit_autocast.py
|
Revert "[JIT] Support scripting torch.is_autocast_enabled() (#81305)"
|
2022-07-28 00:02:53 +00:00 |
|
test_jit_cuda_fuser.py
|
Supports symbolic diff for silu (#81724)
|
2022-08-09 01:18:10 +00:00 |
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
Supports symbolic diff for silu (#81724)
|
2022-08-09 01:18:10 +00:00 |
|
test_jit_legacy.py
|
|
|
|
test_jit_llga_fuser.py
|
|
|
|
test_jit_profiling.py
|
|
|
|
test_jit_simple.py
|
|
|
|
test_jit_string.py
|
|
|
|
test_jiterator.py
|
|
|
|
test_kernel_launch_checks.py
|
|
|
|
test_license.py
|
|
|
|
test_linalg.py
|
Make M1 tests green (#82213)
|
2022-08-05 16:12:08 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
|
|
|
test_meta.py
|
Allow map location to meta device (#82603)
|
2022-08-08 19:56:59 +00:00 |
|
test_metal.py
|
|
|
|
test_mkl_verbose.py
|
[RFC] enable oneMKL&oneDNN on-demands verbose functinality (#63212)
|
2022-07-27 23:29:35 +00:00 |
|
test_mkldnn.py
|
check params shape for mkldnn_convolution (#76526)
|
2022-08-02 17:26:36 +00:00 |
|
test_mkldnn_verbose.py
|
[RFC] enable oneMKL&oneDNN on-demands verbose functinality (#63212)
|
2022-07-27 23:29:35 +00:00 |
|
test_mobile_optimizer.py
|
|
|
|
test_model_dump.py
|
|
|
|
test_module_init.py
|
Add prelu op and module for quantized CPU backend (#73491)
|
2022-07-20 07:48:15 +00:00 |
|
test_modules.py
|
Fix ModuleInfo skip logic (#80471)
|
2022-07-08 19:52:15 +00:00 |
|
test_monitor.py
|
|
|
|
test_mps.py
|
[MPS] Fix scatter for bool type (#82685)
|
2022-08-03 14:54:47 +00:00 |
|
test_multiprocessing.py
|
|
|
|
test_multiprocessing_spawn.py
|
|
|
|
test_namedtensor.py
|
Enable dim=None for torch.var (#82765)
|
2022-08-04 20:47:27 +00:00 |
|
test_namedtuple_return_api.py
|
[Refactoring] making the code more Pythonic (#82929)
|
2022-08-08 06:08:21 +00:00 |
|
test_native_functions.py
|
|
|
|
test_native_mha.py
|
|
|
|
test_nestedtensor.py
|
Add nested tensor contiguous (#82147)
|
2022-08-09 01:51:37 +00:00 |
|
test_nn.py
|
Resolve TODO for GroupNorm numerical issues (#82423)
|
2022-08-03 19:42:26 +00:00 |
|
test_nnapi.py
|
Skip TestNNAPI tests if QNNPACK is not supported (#82882)
|
2022-08-05 16:19:06 +00:00 |
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
|
|
|
test_openmp.py
|
|
|
|
test_ops.py
|
Allow ufunc OpInfos to have no reference (#82348)
|
2022-08-09 04:38:17 +00: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
|
Reenable slow gradcheck and make it pass (#80514)
|
2022-07-22 02:05:37 +00:00 |
|
test_optim.py
|
Handle complex optimization in Adamax by treating complex numbers as 2D real numbers (#80319)
|
2022-08-05 21:03:18 +00:00 |
|
test_overrides.py
|
Add EnableTorchFunction (#82647)
|
2022-08-07 03:04:46 +00:00 |
|
test_package.py
|
|
|
|
test_per_overload_api.py
|
|
|
|
test_prims.py
|
Add torch.ops.nvprims namespace for nvFuser-specific prims (#82155)
|
2022-08-04 16:51:56 +00:00 |
|
test_profiler.py
|
[PyTorch] Proper reset execution graph data in remove callback registration (#82910)
|
2022-08-08 21:14:58 +00:00 |
|
test_profiler_tree.py
|
Make M1 tests green (#82213)
|
2022-08-05 16:12:08 +00:00 |
|
test_proxy_tensor.py
|
[composite compliance] quantile and nanquantile (#81767)
|
2022-08-08 14:42:51 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
Rename SymbolicIntNode to SymIntNodeImpl (#82350)
|
2022-07-28 18:27:45 +00:00 |
|
test_python_dispatch.py
|
Make python TensorOption signatures consistent with JIT schemas (#82241)
|
2022-08-07 00:10:27 +00:00 |
|
test_pytree.py
|
|
|
|
test_quantization.py
|
[Quant][fx] Implement BackendConfig (part 1) (#81469)
|
2022-07-24 00:34:48 +00:00 |
|
test_reductions.py
|
Enable dim=None for torch.var (#82765)
|
2022-08-04 20:47:27 +00:00 |
|
test_scatter_gather_ops.py
|
|
|
|
test_segment_reductions.py
|
|
|
|
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
|
|
|
|
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
|
|
|
|
test_sort_and_select.py
|
|
|
|
test_sparse.py
|
Revert "Implement mul(dense, sparse), mul(sparse, dense) for sparse COO tensors. (#81556)"
|
2022-08-05 19:39:01 +00:00 |
|
test_sparse_csr.py
|
Fix BSR->Dense Batched Bug (#82120)
|
2022-08-06 02:24:20 +00:00 |
|
test_spectral_ops.py
|
Fix istft default output length (#80031)
|
2022-07-08 23:43:29 +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
|
|
|
|
test_tensor_creation_ops.py
|
Fix a segfault in new_empty_strided (#82422)
|
2022-08-04 22:01:18 +00:00 |
|
test_tensorboard.py
|
|
|
|
test_tensorexpr.py
|
|
|
|
test_tensorexpr_pybind.py
|
|
|
|
test_testing.py
|
Move OpInfo class into new opinfo folder (#82540)
|
2022-08-05 15:10:17 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
Add nondeterministic alert to CUDA cumsum (#75693)
|
2022-08-04 01:58:29 +00:00 |
|
test_transformers.py
|
Initial private SDP interface and naive composite impl (#81956)
|
2022-08-01 22:26:18 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
|
|
|
test_type_promotion.py
|
Revert "Implement mul(dense, sparse), mul(sparse, dense) for sparse COO tensors. (#81556)"
|
2022-08-05 19:39:01 +00:00 |
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
Allow ufunc OpInfos to have no reference (#82348)
|
2022-08-09 04:38:17 +00:00 |
|
test_utils.py
|
Prevent ref cycle creation in inner hook (#82776)
|
2022-08-06 00:31:22 +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
|
|
|
|
test_vulkan.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
