| .. |
|
_nvfuser
|
|
|
|
ao/sparsity
|
Remove commented out pdb (#101993)
|
2023-05-23 02:20:10 +00:00 |
|
autograd
|
|
|
|
backends/xeon
|
|
|
|
benchmark_utils
|
Add asan slow test shard (#99925)
|
2023-04-26 21:10:55 +00:00 |
|
bottleneck_test
|
|
|
|
cpp
|
Segmentation fault in flatbuffers when parsing malformed modules (#95221)
|
2023-05-24 21:16:19 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
Extend storage create for custom storageImpl (#100237)
|
2023-05-17 04:30:13 +00:00 |
|
custom_backend
|
|
|
|
custom_operator
|
|
|
|
distributed
|
[Composable] fully_shard load_optim test (#102692)
|
2023-06-04 18:31:22 +00:00 |
|
distributions
|
|
|
|
dynamo
|
Revert "torch.compiler public namespace (#102182)"
|
2023-06-05 06:52:37 +00:00 |
|
edge
|
|
|
|
error_messages
|
|
|
|
expect
|
Add missing decompositons/lowerings for logical/bitwise operators (#102566)
|
2023-06-02 14:27:17 +00:00 |
|
export
|
[export] Replicate exportdb examples and tests in oss. (#102769)
|
2023-06-04 20:01:57 +00:00 |
|
forward_backward_compatibility
|
support ConvBinaryInplace in Inductor cpp wrapper (#101394)
|
2023-06-01 00:22:29 +00:00 |
|
functorch
|
[pt2] add meta for _linalg_slogdet (#102464)
|
2023-06-05 03:17:08 +00:00 |
|
fx
|
[fx] Better replacements finder in subgraph rewriter (#100556)
|
2023-05-16 14:08:44 +00:00 |
|
inductor
|
Revert "Replace _dynamo.config with an object instead of module (#96455)"
|
2023-06-05 07:06:14 +00:00 |
|
jit
|
fix bug in trace model when out-operator has more than one output (#101563)
|
2023-05-31 19:39:52 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
Use yaml.SafeLoader instead of legacy yaml.Loader (#100443)
|
2023-05-02 18:32:36 +00:00 |
|
mobile
|
|
|
|
nn
|
update tf32 thresholds in nn/test_convolution.py (#102015)
|
2023-05-24 22:42:25 +00:00 |
|
onnx
|
[ONNX] Support aten::logit (#102377)
|
2023-06-02 03:39:35 +00:00 |
|
onnx_caffe2
|
|
|
|
optim
|
Disallow _foreach_utils.py, but allow it to be inlined (#102221)
|
2023-06-02 05:14:09 +00:00 |
|
package
|
Adding serialization ID to inline container (#100994)
|
2023-05-17 17:08:48 +00:00 |
|
profiler
|
[profiler] Reintroduce forward-backward links (#102424)
|
2023-05-31 02:50:38 +00:00 |
|
quantization
|
[quant][pt2e][be] Remove _input_output_share_observers and _reuse_input_obs_or_fq from QuantizationAnnotation (#102854)
|
2023-06-03 07:31:09 +00:00 |
|
scripts
|
Fix flake8 lint errors reported by ruff - take 2 (#99798)
|
2023-04-23 23:09:51 +00:00 |
|
test_img
|
|
|
|
typing
|
|
|
|
_test_bazel.py
|
[bazel] add build for functorch (#101475)
|
2023-05-18 20:29:08 +00:00 |
|
allowlist_for_publicAPI.json
|
|
|
|
conftest.py
|
Do not collect and skip non-disabled tests when rerunning disabled tests (#102107)
|
2023-05-27 12:10:36 +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
|
Add print statements to debug sharding error (#102713)
|
2023-06-01 22:38:28 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
|
|
|
test_autocast.py
|
fix cpu autocast check in rnn (#100621)
|
2023-05-09 04:34:18 +00:00 |
|
test_autograd.py
|
Remove incorrect THP{Cpp,}Function_traverse PyObject traversals (#102860)
|
2023-06-02 22:05:25 +00:00 |
|
test_binary_ufuncs.py
|
Fixed indentation error in test_binary_ufuncs.py (#102244)
|
2023-05-25 21:21:30 +00:00 |
|
test_bundled_images.py
|
|
|
|
test_bundled_inputs.py
|
|
|
|
test_comparison_utils.py
|
|
|
|
test_compile_benchmark_util.py
|
|
|
|
test_complex.py
|
|
|
|
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
|
|
|
|
test_cpp_extensions_jit.py
|
|
|
|
test_cpp_extensions_open_device_registration.py
|
make device normalization more generic in faketensor (#102519)
|
2023-06-04 01:44:21 +00:00 |
|
test_cuda.py
|
[BE] Make test_no_triton_on_import simple (#102674)
|
2023-06-01 20:31:18 +00:00 |
|
test_cuda_expandable_segments.py
|
|
|
|
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
|
[codemod][3.10][NamedTuple] Use typing_extensions to get NamedTuple Generics (#101830)
|
2023-05-19 22:50:18 +00:00 |
|
test_decomp.py
|
disable tf32 for rnn tests and norm tests (#102005)
|
2023-05-24 02:22:58 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
|
|
|
test_dlpack.py
|
add additional stream priority for cuda streams (#101956)
|
2023-05-27 02:36:16 +00:00 |
|
test_dynamic_shapes.py
|
group constraints by arg (#102096)
|
2023-05-24 05:27:54 +00:00 |
|
test_expanded_weights.py
|
disable tf32 for rnn tests and norm tests (#102005)
|
2023-05-24 02:22:58 +00:00 |
|
test_fake_tensor.py
|
[inductor] layout optimization for conv (#99773)
|
2023-06-02 21:08:18 +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
|
Add forward mode AD to in-place foreach functions (#100695)
|
2023-05-25 15:39:48 +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
|
Enable mypy allow redefinition (#102046)
|
2023-05-24 07:05:30 +00:00 |
|
test_fx.py
|
[fx][2/n] Add metadata to placeholders (#102195)
|
2023-05-25 07:04:20 +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
|
|
|
|
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
|
|
|
|
test_kernel_launch_checks.py
|
|
|
|
test_legacy_vmap.py
|
|
|
|
test_license.py
|
|
|
|
test_linalg.py
|
Enable hipSOLVER in ROCm builds (#97370)
|
2023-05-31 16:53:23 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
|
|
|
test_maskedtensor.py
|
Tighten FakeTensor reentrancy asserts, add debugging (#102091)
|
2023-05-24 05:37:51 +00:00 |
|
test_matmul_cuda.py
|
|
|
|
test_meta.py
|
[pt2] add SymInt support for linalg.vander (#102469)
|
2023-06-04 09:58:02 +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
|
support ConvBinaryInplace in Inductor cpp wrapper (#101394)
|
2023-06-01 00:22:29 +00:00 |
|
test_mkldnn_verbose.py
|
|
|
|
test_mobile_optimizer.py
|
|
|
|
test_model_dump.py
|
|
|
|
test_module_init.py
|
|
|
|
test_modules.py
|
test_memory_format fix for test_modules.py (#102006)
|
2023-05-24 02:32:45 +00:00 |
|
test_monitor.py
|
|
|
|
test_mps.py
|
[MPS] Fix softplus with f16 input (#101948)
|
2023-05-31 00:40:10 +00:00 |
|
test_multiprocessing.py
|
|
|
|
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
|
add function specializations for the case of parameters in BFloat16 data type (#100233)
|
2023-05-31 02:01:07 +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
|
[inductor] Added smooth_l1_loss refs (#102077)
|
2023-05-24 15:07:08 +00:00 |
|
test_ops_fwd_gradients.py
|
|
|
|
test_ops_gradients.py
|
[custom_op] explicit autograd API (#101824)
|
2023-05-23 18:31:29 +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
|
|
|
|
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
|
[pt2] add meta for _linalg_slogdet (#102464)
|
2023-06-05 03:17:08 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
|
|
|
test_python_dispatch.py
|
Add API to construct the functional variant of an op (#102293)
|
2023-06-02 13:36:50 +00:00 |
|
test_pytree.py
|
Serialize pytree to string v2 (#102708)
|
2023-06-01 19:51:28 +00:00 |
|
test_quantization.py
|
[reland][quant][test] Fix broken PT2 import, add warnings (#102819)
|
2023-06-02 22:35:30 +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
|
Preserve coalesce state in sparse COO tensor serialization (#102647)
|
2023-06-03 01:37:52 +00:00 |
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
|
|
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
Make 1D integer sorting work in parallel (#100081)
|
2023-06-02 07:41:28 +00:00 |
|
test_sparse.py
|
torch.sparse.softmax: allow negative dim (#102172)
|
2023-05-24 19:43:47 +00:00 |
|
test_sparse_csr.py
|
Skip test test_triton_bsr_dense_bmm if not TEST_WITH_TORCHINDUCTOR [v2] (#102660)
|
2023-06-01 20:36:45 +00:00 |
|
test_spectral_ops.py
|
|
|
|
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
|
Introduce fast path for cuda_equal (#102714)
|
2023-06-03 05:49:49 +00:00 |
|
test_transformers.py
|
Move static checks of layers[0] (e.g., isinstance check) to model build time (#102045)
|
2023-05-30 19:42:01 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
|
|
|
test_type_promotion.py
|
|
|
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
add Half support for logsigmoid, threshold, elu, gelu, hardtanh, hardsigmoid, hardswish, hardshrink, softshrink, leakyrelu, softplus, glu, silu, mish, and prelu on CPU (#98745)
|
2023-05-27 16:20:21 +00:00 |
|
test_utils.py
|
Add torch._utils.render_call, improve printoptions (#102623)
|
2023-05-31 22:08:04 +00:00 |
|
test_view_ops.py
|
|
|
|
test_vulkan.py
|
|
|
|
test_weak.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
