saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test
History
Driss Guessous 81bbee7d7e [SDPA] Adds basic correctness checks (#94274) 
# Summary
Add more checks around shape constraints as well as update the sdp_utils to properly catch different head_dims between qk and v for flash_attention which is not supported.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/94274
Approved by: https://github.com/cpuhrsch
2023-02-09 08:05:26 +00:00
..
ao/sparsity AO migration: replace torch internal callsites (#94170) 2023-02-07 02:32:23 +00:00
autograd
backends/xeon
benchmark_utils
bottleneck_test
cpp [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
cpp_api_parity
cpp_extensions update aten op overload to not use from to avoid compile errors (#89797) 2023-02-08 22:04:59 +00:00
custom_backend Migrate PyTorch to C++17 (#85969) 2022-12-08 02:27:48 +00:00
custom_operator Migrate PyTorch to C++17 (#85969) 2022-12-08 02:27:48 +00:00
distributed [Inductor] added aten.geometric_ decomp (#91672) 2023-02-09 07:29:14 +00:00
distributions [BE] Merge isinstance calls together (#94419) 2023-02-09 00:47:26 +00:00
dynamo Decouple PT2 dynamic shapes from the functorch setting (#94469) 2023-02-09 06:41:41 +00:00
edge [torchgen] Add CI job to cover custom ops registration for Executorch (#91291) 2023-01-14 02:30:54 +00:00
error_messages
expect [Inductor] added aten.geometric_ decomp (#91672) 2023-02-09 07:29:14 +00:00
forward_backward_compatibility update aten op overload to not use from to avoid compile errors (#89797) 2023-02-08 22:04:59 +00:00
functorch [decompositions] add decomposition for RNN with packed sequence (#91281) 2023-02-08 14:16:30 +00:00
fx [BE]: Apply pyupgrade yield from and unit test alias upgrades (#94309) 2023-02-07 20:08:58 +00:00
inductor [Inductor] added aten.geometric_ decomp (#91672) 2023-02-09 07:29:14 +00:00
jit update aten op overload to not use from to avoid compile errors (#89797) 2023-02-08 22:04:59 +00:00
jit_hooks Migrate PyTorch to C++17 (#85969) 2022-12-08 02:27:48 +00:00
lazy Fix exception causes all over the codebase (#90271) 2022-12-07 04:29:00 +00:00
mobile AO migration: replace torch internal callsites (#94170) 2023-02-07 02:32:23 +00:00
nn fix: max_unpool3d buffer overflow (#94372) 2023-02-08 19:48:25 +00:00
onnx [ONNX] Use optional op to keep None in results for ONNX internal tests (#84789) 2023-02-08 23:04:47 +00:00
onnx_caffe2 [BE] Merge isinstance calls together (#94419) 2023-02-09 00:47:26 +00:00
package [package] Add better debugging for torch.package (#92939) 2023-01-26 02:11:12 +00:00
profiler [nn] zero_grad() set_to_none default True (#92731) 2023-01-26 01:04:28 +00:00
quantization [BE] Merge isinstance calls together (#94419) 2023-02-09 00:47:26 +00:00
scripts Fix typos in messages under test (#89121) 2022-11-17 01:55:03 +00:00
test_img
typing
allowlist_for_publicAPI.json temp fix for segment reduce undocumented FC window (#94242) 2023-02-07 18:27:01 +00:00
conftest.py
create_dummy_torchscript_model.py
delete.py
HowToWriteTestsUsingFileCheck.md
linear.py
load_torchscript_model.py
mkl_verbose.py
mkldnn_verbose.py
run_doctests.sh Enable xdoctest runner in CI for real this time (#83816) 2022-12-29 05:32:42 +00:00
run_test.py AO migration: replace torch internal callsites (#94170) 2023-02-07 02:32:23 +00:00
simulate_nccl_errors.py
test_ao_sparsity.py [pruning][core][feature] Add in SaliencyPruner to pruner._experimental (#91814) 2023-01-10 04:04:55 +00:00
test_autocast.py Revert "remove torch.equal usages (#89527)" 2022-12-02 21:36:13 +00:00
test_autograd.py add torch.autograd._set_view_replay_enabled, use in aot autograd (#92588) 2023-02-08 01:48:32 +00:00
test_binary_ufuncs.py Vectorized more stable complex division (#93277) 2023-02-03 11:48:20 +00:00
test_bundled_images.py
test_bundled_inputs.py
test_comparison_utils.py
test_complex.py
test_cpp_api_parity.py
test_cpp_extensions_aot.py Set min supported Python version to 3.8 (#93155) 2023-01-29 18:28:46 +00:00
test_cpp_extensions_jit.py Fix test_jit_cuda_archflags on Windows (#93332) 2023-02-02 02:49:27 +00:00
test_cpp_extensions_open_device_registration.py
test_cuda.py Look up group["capturable"], not defaults["capturable"] in Adam(W) (#94149) 2023-02-07 00:24:35 +00:00
test_cuda_nvml_based_avail.py
test_cuda_primary_ctx.py
test_cuda_sanitizer.py
test_cuda_trace.py
test_dataloader.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_datapipe.py [BE]: Apply pyupgrade yield from and unit test alias upgrades (#94309) 2023-02-07 20:08:58 +00:00
test_decomp.py [decompositions] GRU decompositon with and without packed sequence (#91466) 2023-02-08 14:16:30 +00:00
test_deploy.py
test_determination.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_dispatch.py
test_dlpack.py
test_dynamic_shapes.py Track and record hint on SymNode and use when possible (#94201) 2023-02-09 00:00:44 +00:00
test_expanded_weights.py [decompositions] GRU decompositon with and without packed sequence (#91466) 2023-02-08 14:16:30 +00:00
test_fake_tensor.py update aten op overload to not use from to avoid compile errors (#89797) 2023-02-08 22:04:59 +00:00
test_foreach.py [mta] Backward of unary foreach functions (#89591) 2023-01-23 08:28:06 +00:00
test_function_schema.py [frontend] Expose real_type getter for torch.Argument (#91938) 2023-01-12 01:26:50 +00:00
test_functional_autograd_benchmark.py
test_functional_optim.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_functionalization.py [Reland] Refactor stack_trace preservation for node meta preservation (#90803) (#92400) 2023-01-30 23:30:43 +00:00
test_futures.py
test_fx.py Revert "feat(fx): make_fx should be aware of functions wrapped with @fx.wrap (#93273)" 2023-02-09 03:33:09 +00:00
test_fx_experimental.py [fx][passes] Implement annotate getitem node FX passes (#90237) 2022-12-06 23:18:55 +00:00
test_fx_passes.py refactor the dfs cyclic search from recursion to iterative approach (#91042) 2022-12-20 23:15:30 +00:00
test_fx_reinplace_pass.py *_scatter ops should preserve input stride/storage_offset (#91029) 2022-12-22 19:41:53 +00:00
test_hub.py
test_import_stats.py
test_indexing.py Add a mode to rerun all disabled tests (without running anything else) (#88646) 2022-11-15 05:08:26 +00:00
test_itt.py
test_jit.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_jit_autocast.py prepare removal of deprecated functionality in torch.testing (#87969) 2022-11-02 14:04:48 +00:00
test_jit_cuda_fuser.py [CUDA] Drop CUDA < 11.0 test flags (#92605) 2023-01-24 04:34:06 +00:00
test_jit_disabled.py
test_jit_fuser.py
test_jit_fuser_legacy.py
test_jit_fuser_te.py Revert "Disable torch_jit_fuser_te for dynamo CI (#92945)" 2023-01-26 21:41:17 +00:00
test_jit_legacy.py
test_jit_llga_fuser.py Set correct size of attr::output_layouts when the graph has multiple outputs in JIT oneDNN fuser (#88496) 2022-11-15 07:29:55 +00:00
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_jiterator.py
test_kernel_launch_checks.py
test_legacy_vmap.py Remove deprecated torch.symeig (#70988) 2023-01-31 11:59:11 +00:00
test_license.py
test_linalg.py [BE] Add empty tensor check to _compute_linear_combination (#94245) 2023-02-07 11:31:11 +00:00
test_logging.py
test_masked.py Support per-parameter test decoration (#91658) 2023-01-04 21:08:32 +00:00
test_maskedtensor.py
test_matmul_cuda.py [CUDA] Drop CUDA < 11.0 test flags (#92605) 2023-01-24 04:34:06 +00:00
test_meta.py [BE]: Apply pyupgrade yield from and unit test alias upgrades (#94309) 2023-02-07 20:08:58 +00:00
test_metal.py
test_mkl_verbose.py
test_mkldnn.py Optimize ConvTransposed2D with mkldnn float32 and bfloat16 on CPU (#92530) 2023-02-06 10:11:25 +00:00
test_mkldnn_fusion.py [Re-open 90266] [inductor] weight prepack for _convolution_transpose_pointwise (#91955) 2023-01-31 13:28:57 +00:00
test_mkldnn_verbose.py
test_mobile_optimizer.py [Reland] Clean Up MobileOptimizerType Rewrite Flags Public API and Documentation (#92081) 2023-01-14 17:06:00 +00:00
test_model_dump.py
test_module_init.py AO migration: replace torch internal callsites (#94170) 2023-02-07 02:32:23 +00:00
test_modules.py Revert "Update Module.__setattr__ to respect property setters (#92044)" 2023-01-21 02:39:21 +00:00
test_monitor.py
test_mps.py [MPS] Add 2d grid sampler (#94273) 2023-02-09 02:25:46 +00:00
test_multiprocessing.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_multiprocessing_spawn.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_namedtensor.py Revert "remove torch.equal usages (#89527)" 2022-12-02 21:36:13 +00:00
test_namedtuple_return_api.py [SDPA] Wire up FlashAttention's backward (#92917) 2023-02-02 04:02:30 +00:00
test_native_functions.py Improve argument printing (#87601) 2022-10-24 23:55:10 +00:00
test_native_mha.py Use scaled_dot_product_attention within attention.cpp (#87312) 2022-10-31 04:06:31 +00:00
test_nestedtensor.py [NestedTensor] Call contiguous in linear backward (#94317) 2023-02-07 23:43:46 +00:00
test_nn.py Fix SIGFPE in slow_conv3d_forward_out_cpu (#94325) 2023-02-08 14:15:39 +00:00
test_nnapi.py
test_numba_integration.py
test_numpy_interop.py
test_nvfuser_dynamo.py Refactor dynamo training backends (#93409) 2023-02-03 03:07:15 +00:00
test_nvfuser_frontend.py [NVFUSER] refactor nvfuser build (#89621) 2023-01-26 02:50:44 +00:00
test_openmp.py
test_ops.py Make segment_reduce properly private. (#93166) 2023-02-06 18:32:23 +00:00
test_ops_fwd_gradients.py Mitigate flaky test_ops_fwd_gradients on macOS (#89410) 2022-11-22 00:13:38 +00:00
test_ops_gradients.py Split out forward AD tests from test_ops_gradients and reenable slow gradcheck CI (#88216) 2022-11-03 00:20:45 +00:00
test_ops_jit.py Split out forward AD tests from test_ops_gradients and reenable slow gradcheck CI (#88216) 2022-11-03 00:20:45 +00:00
test_optim.py [BE] Merge isinstance calls together (#94419) 2023-02-09 00:47:26 +00:00
test_overrides.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_package.py
test_per_overload_api.py
test_prims.py [NVFUSER] refactor nvfuser build (#89621) 2023-01-26 02:50:44 +00:00
test_proxy_tensor.py Make segment_reduce properly private. (#93166) 2023-02-06 18:32:23 +00:00
test_pruning_op.py
test_public_bindings.py [jit] Support Awaitable type (#90863) 2023-01-30 17:38:59 +00:00
test_python_dispatch.py [BE]: Apply pyupgrade yield from and unit test alias upgrades (#94309) 2023-02-07 20:08:58 +00:00
test_pytree.py Make nested TreeSpec printing nicer (#46538) (#86546) 2022-10-18 16:50:39 +00:00
test_quantization.py [quant][pt2e] Add a test to confirm we can set qconfig according to module_name (#91977) 2023-01-12 21:59:02 +00:00
test_reductions.py Logcumsumexp for CPU (#93153) 2023-01-27 22:29:33 +00:00
test_scatter_gather_ops.py optimize gather performance for gnn usage on CPU (#87586) 2023-01-12 00:43:43 +00:00
test_schema_check.py Skip some unittests (#90609) 2022-12-11 23:53:05 +00:00
test_segment_reductions.py Make segment_reduce properly private. (#93166) 2023-02-06 18:32:23 +00:00
test_serialization.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py Added vectorized horizontal flip path for channels last for NcHW (#91806) 2023-01-23 20:15:30 +00:00
test_show_pickle.py
test_sort_and_select.py fix sort crash when the input is expanded scalar (#91752) 2023-01-09 02:02:56 +00:00
test_sparse.py Add CSR->BSC and CSC->BSR conversions (#93301) 2023-02-07 19:22:05 +00:00
test_sparse_csr.py Add CSR->BSC and CSC->BSR conversions (#93301) 2023-02-07 19:22:05 +00:00
test_spectral_ops.py
test_stateless.py [functorch][nn] Refactor NN stateless APIs by swapping module tensors (#92536) 2023-02-08 17:31:38 +00:00
test_static_runtime.py
test_subclass.py
test_tensor_creation_ops.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_tensorboard.py use proper temp directories in test_tensorboard.py (#89826) 2022-12-09 19:33:03 +00:00
test_tensorexpr.py Revert "remove torch.equal usages (#89527)" 2022-12-02 21:36:13 +00:00
test_tensorexpr_pybind.py
test_testing.py [BE]: Apply pyupgrade yield from and unit test alias upgrades (#94309) 2023-02-07 20:08:58 +00:00
test_throughput_benchmark.py
test_torch.py [Dynamo] Support torch.Tensor.fn as TorchVariable, not UserDefinedObjectVariable, preventing graph break (#93243) 2023-02-07 09:26:50 +00:00
test_transformers.py [SDPA] Adds basic correctness checks (#94274) 2023-02-09 08:05:26 +00:00
test_type_hints.py
test_type_info.py
test_type_promotion.py [Dynamo] Support torch.get_default_dtype (#89790) 2022-12-19 04:14:11 +00:00
test_typing.py
test_unary_ufuncs.py Log1p complex for CUDA (#90422) 2022-12-09 19:53:22 +00:00
test_utils.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_view_ops.py Deprecate .mT,.T,.mH,.H on 0D tensors (#92143) 2023-01-17 16:54:35 +00:00
test_vulkan.py
test_weak.py [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
test_xnnpack_integration.py disable test that times out in fbcode (#88758) 2022-11-10 00:28:02 +00:00