| .. |
|
_nvfuser
|
|
|
|
ao/sparsity
|
Back out "[core][pruning][be] rename BaseSparsifier to BasePruner (#98747)" (#99171)
|
2023-04-15 00:37:45 +00:00 |
|
autograd
|
|
|
|
backends/xeon
|
|
|
|
benchmark_utils
|
Add asan slow test shard (#99925)
|
2023-04-26 21:10:55 +00:00 |
|
bottleneck_test
|
|
|
|
cpp
|
Switch AOT Inductor test to export, add dynamic, fix invocation bug (#101585)
|
2023-05-17 05:52:08 +00:00 |
|
cpp_api_parity
|
[BE] Remove unnecessary dict comprehensions (#97116)
|
2023-03-20 00:56:57 +00:00 |
|
cpp_extensions
|
Extend storage create for custom storageImpl (#100237)
|
2023-05-17 04:30:13 +00:00 |
|
custom_backend
|
|
|
|
custom_operator
|
Enable TestTorchbind on Windows (#96507)
|
2023-03-16 16:18:08 +00:00 |
|
distributed
|
Enables barrier to support the specified device (#99589)
|
2023-05-17 05:26:04 +00:00 |
|
distributions
|
|
|
|
dynamo
|
[dynamo] Activation checkpoint higher order ops - Reland 101028 (#101790)
|
2023-05-18 19:09:14 +00:00 |
|
edge
|
|
|
|
error_messages
|
|
|
|
expect
|
[inductor/decomp] Add aten._unsafe_index to disable range checks (#101602)
|
2023-05-17 23:36:24 +00:00 |
|
export
|
Actually functionalize torch.export (#101433)
|
2023-05-17 05:09:24 +00:00 |
|
forward_backward_compatibility
|
Upstream xformers code (#100583)
|
2023-05-18 16:15:34 +00:00 |
|
functorch
|
Support map autograd and pytree in/out. (#101633)
|
2023-05-17 16:52:26 +00:00 |
|
fx
|
[fx] Better replacements finder in subgraph rewriter (#100556)
|
2023-05-16 14:08:44 +00:00 |
|
inductor
|
inductor: skip weight packing when has zero shape (#101355)
|
2023-05-18 03:32:45 +00:00 |
|
jit
|
[BE]: enable PLE error codes in ruff and fix bugs (#101079)
|
2023-05-11 23:57:25 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
Use yaml.SafeLoader instead of legacy yaml.Loader (#100443)
|
2023-05-02 18:32:36 +00:00 |
|
mobile
|
|
|
|
nn
|
Thread generator through trunc_normal_ (#100810)
|
2023-05-12 01:04:08 +00:00 |
|
onnx
|
[ONNX] Diagnostic 'log' and 'log_and_raise_if_error' (#100407)
|
2023-05-16 04:55:50 +00:00 |
|
onnx_caffe2
|
|
|
|
optim
|
Handle tail 0-size tensor appropriately in MultiTensorApply (#100811)
|
2023-05-12 20:26:45 +00:00 |
|
package
|
Adding serialization ID to inline container (#100994)
|
2023-05-17 17:08:48 +00:00 |
|
profiler
|
[profiler] When record_inputs=True, record scalar lists of length <= 30 (#100593)
|
2023-05-16 07:58:46 +00:00 |
|
quantization
|
[quant][pt2] Handle maxpool + conv + bn case in prepare QAT (#100941)
|
2023-05-17 17:36:32 +00:00 |
|
scripts
|
Fix flake8 lint errors reported by ruff - take 2 (#99798)
|
2023-04-23 23:09:51 +00:00 |
|
test_img
|
|
|
|
typing
|
|
|
|
allowlist_for_publicAPI.json
|
|
|
|
conftest.py
|
File level rerun changes (#100200)
|
2023-04-28 20:57:49 +00:00 |
|
create_dummy_torchscript_model.py
|
|
|
|
delete.py
|
|
|
|
HowToWriteTestsUsingFileCheck.md
|
|
|
|
linear.py
|
|
|
|
load_torchscript_model.py
|
|
|
|
mkl_verbose.py
|
|
|
|
mkldnn_verbose.py
|
|
|
|
run_doctests.sh
|
|
|
|
run_test.py
|
Check for pytest extensions in run_test (#100916)
|
2023-05-17 20:27:55 +00:00 |
|
simulate_nccl_errors.py
|
Fix G001,G002,G003 in logs to % syntax (#97812)
|
2023-04-01 01:43:33 +00:00 |
|
test_ao_sparsity.py
|
Back out "[core][pruning][be] rename BaseSparsifier to BasePruner (#98747)" (#99171)
|
2023-04-15 00:37:45 +00:00 |
|
test_autocast.py
|
fix cpu autocast check in rnn (#100621)
|
2023-05-09 04:34:18 +00:00 |
|
test_autograd.py
|
[BE] Make some simplifications to torch.utils.checkpoint logic (#101193)
|
2023-05-12 04:35:22 +00:00 |
|
test_bazel.py
|
[bazel] add python targets (#101003)
|
2023-05-12 19:44:01 +00:00 |
|
test_binary_ufuncs.py
|
[Dynamo] Fix nested function resume execution (#100426)
|
2023-05-11 03:10:23 +00:00 |
|
test_bundled_images.py
|
|
|
|
test_bundled_inputs.py
|
|
|
|
test_comparison_utils.py
|
Add missing __main__ in two unittests (#97302)
|
2023-03-22 19:09:08 +00:00 |
|
test_compile_benchmark_util.py
|
torch.compile benchmark utility (#97699)
|
2023-04-12 03:02:06 +00:00 |
|
test_complex.py
|
Fix CPU vectorized eq and ne operations for complex types (#97374)
|
2023-04-14 02:02:16 +00:00 |
|
test_content_store.py
|
Add load_storage (#100519)
|
2023-05-05 05:25:03 +00:00 |
|
test_cpp_api_parity.py
|
|
|
|
test_cpp_extensions_aot.py
|
Support large negative SymInt (#99157)
|
2023-04-15 22:43:51 +00:00 |
|
test_cpp_extensions_jit.py
|
|
|
|
test_cpp_extensions_open_device_registration.py
|
Extend storage create for custom storageImpl (#100237)
|
2023-05-17 04:30:13 +00:00 |
|
test_cuda.py
|
Run test_cpp_memory_snapshot_pickle only when linux and x86_64 (#101366)
|
2023-05-17 19:44:21 +00:00 |
|
test_cuda_expandable_segments.py
|
Revert "Revert "Expandable blocks in allocator (#96995)"" (#99275)
|
2023-04-17 23:46:08 +00:00 |
|
test_cuda_nvml_based_avail.py
|
|
|
|
test_cuda_primary_ctx.py
|
|
|
|
test_cuda_sanitizer.py
|
|
|
|
test_cuda_trace.py
|
|
|
|
test_dataloader.py
|
[DataLoader] Adding StackDataset (#101338)
|
2023-05-18 00:57:12 +00:00 |
|
test_datapipe.py
|
[DataPipe] Update docstring for functional form of DataPipes (#100446)
|
2023-05-18 19:59:00 +00:00 |
|
test_decomp.py
|
Revert "[opinfo] empty_strided (#100890)"
|
2023-05-17 19:00:15 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
|
|
|
test_dlpack.py
|
|
|
|
test_dynamic_shapes.py
|
remove default lower bound in dynamic_dim suggestions (#101636)
|
2023-05-17 19:55:04 +00:00 |
|
test_expanded_weights.py
|
|
|
|
test_fake_tensor.py
|
CustomOp simple abstract implementation registration (#99439)
|
2023-04-28 13:45:39 +00:00 |
|
test_flop_counter.py
|
tweak heuristic for sdpa selection based off of *data* (and a decision tree) (#99644)
|
2023-04-21 23:28:44 +00:00 |
|
test_foreach.py
|
Enable backward on _foreach_zero_ (#101149)
|
2023-05-17 03:10:13 +00:00 |
|
test_function_schema.py
|
|
|
|
test_functional_autograd_benchmark.py
|
|
|
|
test_functional_optim.py
|
|
|
|
test_functionalization.py
|
functionalization: error during mutations on mem overlap (#99919)
|
2023-04-26 16:39:40 +00:00 |
|
test_functionalization_of_rng_ops.py
|
[philox_rand] Add decomps (#100206)
|
2023-04-28 02:20:13 +00:00 |
|
test_futures.py
|
|
|
|
test_fx.py
|
Revert "[fx] change from #users to num_users in graph printout (#101140)"
|
2023-05-15 14:35:22 +00:00 |
|
test_fx_experimental.py
|
fix conv+bn folding issue for mixed dtype (#99696)
|
2023-04-23 05:13:40 +00:00 |
|
test_fx_passes.py
|
Introduce aggressive merge to CapabilityPartitioner (#100195)
|
2023-05-05 23:20:17 +00:00 |
|
test_fx_reinplace_pass.py
|
|
|
|
test_hub.py
|
torch.hub: add safe weights_only option to load_state_dict_from_url (#98479)
|
2023-04-11 12:44:25 +00:00 |
|
test_import_stats.py
|
|
|
|
test_indexing.py
|
[Dynamo] Fix nested function resume execution (#100426)
|
2023-05-11 03:10:23 +00:00 |
|
test_itt.py
|
|
|
|
test_jit.py
|
Allow C++ custom class to define __repr__ and use it from Python (#100724)
|
2023-05-10 15:46:45 +00:00 |
|
test_jit_autocast.py
|
[BE] Enable C419 rule for any all shortcircuiting (#99890)
|
2023-04-25 15:02:13 +00:00 |
|
test_jit_cuda_fuser.py
|
|
|
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
Fix flake8 lint errors reported by ruff - take 2 (#99798)
|
2023-04-23 23:09:51 +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
|
[ROCm] add skipCUDAIfVersionLessThan to unskip test_jiterator for ROCm (#99197)
|
2023-04-17 16:05:16 +00:00 |
|
test_kernel_launch_checks.py
|
|
|
|
test_legacy_vmap.py
|
|
|
|
test_license.py
|
|
|
|
test_linalg.py
|
Revert "[dynamo 3.11] enable other torch 3.11 dynamo-related tests (#99180)"
|
2023-05-12 16:18:22 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
|
|
|
test_maskedtensor.py
|
fix random mask creation in test_maskedtensor (#97017)
|
2023-03-24 23:55:17 +00:00 |
|
test_matmul_cuda.py
|
[CUBLAS] Specify alignment for cuBlasLt addmm (#98975)
|
2023-04-18 06:19:30 +00:00 |
|
test_meta.py
|
Add aten.searchsorted.Tensor meta kernel (#101637)
|
2023-05-18 06:55:11 +00:00 |
|
test_metal.py
|
|
|
|
test_mkl_verbose.py
|
|
|
|
test_mkldnn.py
|
Revert "fix onednn ConvTranspose2d channels last issue when ic=1 (#99539)"
|
2023-04-21 08:44:28 +00:00 |
|
test_mkldnn_fusion.py
|
|
|
|
test_mkldnn_verbose.py
|
|
|
|
test_mobile_optimizer.py
|
|
|
|
test_model_dump.py
|
|
|
|
test_module_init.py
|
|
|
|
test_modules.py
|
[MPS] Handle MPS failures of test_modules.py in common_modules.py (#95334)
|
2023-05-09 03:55:16 +00:00 |
|
test_monitor.py
|
|
|
|
test_mps.py
|
Revert "[opinfo] empty_strided (#100890)"
|
2023-05-17 19:00:15 +00:00 |
|
test_multiprocessing.py
|
Revert "Revert "Expandable blocks in allocator (#96995)"" (#99275)
|
2023-04-17 23:46:08 +00:00 |
|
test_multiprocessing_spawn.py
|
|
|
|
test_namedtensor.py
|
|
|
|
test_namedtuple_return_api.py
|
Upstream xformers code (#100583)
|
2023-05-18 16:15:34 +00:00 |
|
test_native_functions.py
|
[Dynamo] Fix nested function resume execution (#100426)
|
2023-05-11 03:10:23 +00:00 |
|
test_native_mha.py
|
|
|
|
test_nestedtensor.py
|
Allow discontiguous NestedTensors to empty_like (#98383)
|
2023-05-03 02:27:08 +00:00 |
|
test_nn.py
|
Fix inconsistent torch.nn.MaxPool1d output on cpu and gpu (#99843)
|
2023-05-15 20:27:43 +00:00 |
|
test_nnapi.py
|
Revert "Added round_with_scale_factor arg to ATen (#97868)"
|
2023-04-28 20:47:00 +00:00 |
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
|
|
|
test_nvfuser_dynamo.py
|
|
|
|
test_nvfuser_frontend.py
|
|
|
|
test_openmp.py
|
|
|
|
test_ops.py
|
[opinfo] item (#100313)
|
2023-05-10 11:32:45 +00:00 |
|
test_ops_fwd_gradients.py
|
|
|
|
test_ops_gradients.py
|
Support map autograd and pytree in/out. (#101633)
|
2023-05-17 16:52:26 +00:00 |
|
test_ops_jit.py
|
[BE] Enable C419 rule for any all shortcircuiting (#99890)
|
2023-04-25 15:02:13 +00:00 |
|
test_optim.py
|
[optim][BE] split test file into logical parts: SWA, LR, optim (#101100)
|
2023-05-12 16:41:44 +00:00 |
|
test_overrides.py
|
Make gen_annotated_args support kwargs (#98396)
|
2023-04-06 19:42:26 +00:00 |
|
test_package.py
|
|
|
|
test_per_overload_api.py
|
|
|
|
test_prims.py
|
Fix prims unbind if given dimension size is 0 (#100122)
|
2023-04-26 23:40:21 +00:00 |
|
test_proxy_tensor.py
|
Add aten.searchsorted.Tensor meta kernel (#101637)
|
2023-05-18 06:55:11 +00:00 |
|
test_pruning_op.py
|
Add missing __main__ in two unittests (#97302)
|
2023-03-22 19:09:08 +00:00 |
|
test_public_bindings.py
|
Add Symbool support in python to C++ translation (#98453)
|
2023-04-12 03:21:57 +00:00 |
|
test_python_dispatch.py
|
[reland][CustomOp] Add Dispatcher error callback (#101452)
|
2023-05-16 13:33:31 +00:00 |
|
test_pytree.py
|
|
|
|
test_quantization.py
|
Fix the pt2e UT path after refactor (#99402)
|
2023-04-18 10:48:52 +00:00 |
|
test_reductions.py
|
|
|
|
test_scatter_gather_ops.py
|
[pytorch] Accelerate indexing_backward_kernel with duplicates (#99441 attempt 2) (#100505)
|
2023-05-03 23:52:58 +00:00 |
|
test_schema_check.py
|
Get SchemaCheckMode to error on ops that return inputs directly. Expose as a dynamo backend, eager_debug (#99744)
|
2023-04-27 20:12:42 +00:00 |
|
test_segment_reductions.py
|
Allow data size equal to 0 for SegmentReduce (#99733)
|
2023-04-23 01:59:45 +00:00 |
|
test_serialization.py
|
Support non-ASCII characters in model file paths (#99453)
|
2023-04-26 01:15:49 +00:00 |
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
|
|
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
|
|
|
test_sparse.py
|
Add activation functions (ReLU and SiLU for now) for structured sparse linear operator (#101339)
|
2023-05-18 01:53:18 +00:00 |
|
test_sparse_csr.py
|
sparse compressed validation: allow empty-batched inputs (#101180)
|
2023-05-11 20:30:20 +00:00 |
|
test_spectral_ops.py
|
Revert "[cuda rng] Making offset calculation independent of device properties (#98988)"
|
2023-04-19 17:23:40 +00:00 |
|
test_stateless.py
|
[Functorch] Skip docs setup if called in optimize mode (#100750)
|
2023-05-08 23:36:57 +00:00 |
|
test_static_runtime.py
|
|
|
|
test_subclass.py
|
|
|
|
test_sympy_utils.py
|
Fix a number of issues with divs in ValueRangeAnalysis (#100547)
|
2023-05-04 12:31:55 +00:00 |
|
test_tensor_creation_ops.py
|
Revert "[dynamo 3.11] enable other torch 3.11 dynamo-related tests (#99180)"
|
2023-05-12 16:18:22 +00:00 |
|
test_tensorboard.py
|
[BE] Enable C419 rule for any all shortcircuiting (#99890)
|
2023-04-25 15:02:13 +00:00 |
|
test_tensorexpr.py
|
|
|
|
test_tensorexpr_pybind.py
|
|
|
|
test_testing.py
|
[inductor] Create triton_helpers module for helper functions (#99880)
|
2023-04-27 15:10:50 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
Make interpolate_bilinear deterministic using decomposition (#101115)
|
2023-05-11 22:48:01 +00:00 |
|
test_transformers.py
|
Upstream xformers code (#100583)
|
2023-05-18 16:15:34 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
|
|
|
test_type_promotion.py
|
[BE] Enable flake8-comprehension rule C417 (#97880)
|
2023-03-30 14:34:24 +00:00 |
|
test_typing.py
|
[BE] Enable flake8-comprehension rule C417 (#97880)
|
2023-03-30 14:34:24 +00:00 |
|
test_unary_ufuncs.py
|
|
|
|
test_utils.py
|
Fix torch.utils._traceback on Python 3.11 (#101277)
|
2023-05-14 19:03:16 +00:00 |
|
test_view_ops.py
|
Add overflow check for stride calculation (#94900)
|
2023-04-09 01:30:55 +00:00 |
|
test_vulkan.py
|
|
|
|
test_weak.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
