pytorch/test
Max Ren d2033a0639 [quant][pt2e][xnnpack_quantizer] add support for linear_relu (#117052)
Add support for linear_relu annotation for XNNPACKQuantizer, this allows the input to linear and the output to relu to share the same quantization parameter.s

Differential Revision: [D52574086](https://our.internmc.facebook.com/intern/diff/D52574086/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/117052
Approved by: https://github.com/jerryzh168, https://github.com/digantdesai
2024-01-09 23:19:52 +00:00
..
ao/sparsity [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
autograd Autograd attaches logging hooks only in debug level (#116522) 2024-01-02 20:06:18 +00:00
backends/xeon
benchmark_utils
bottleneck_test
cpp [c10d] Make DebugInfoWriter Singleton across all PG objects (#116489) 2024-01-03 03:42:54 +00:00
cpp_api_parity Add python and C++ support for LPPool3d (#114199) 2023-12-08 18:18:44 +00:00
cpp_extensions Revert "Add function to materialize COW storages (#113396)" 2023-11-20 10:26:01 +00:00
custom_backend
custom_operator Update impl_abstract_pystub to be less boilerplatey (#113182) 2023-11-08 00:39:00 +00:00
distributed [reland] unflatten_tensor on compute stream for DTensorExtension (#117020) 2024-01-09 21:25:15 +00:00
distributions Add bfloat16 CUDA support to binomial distribution (#116932) 2024-01-07 19:50:10 +00:00
dynamo Impl. call_hasattr for BaseUserFunctionVariable (#116049) 2024-01-09 22:58:58 +00:00
edge Adding event_tracer evalue logging calls in codegen (#114584) 2023-11-28 18:32:05 +00:00
error_messages
expect [CPU] Add flash attention mask version (#115913) 2024-01-07 04:58:23 +00:00
export [CPU] Add flash attention mask version (#115913) 2024-01-07 04:58:23 +00:00
forward_backward_compatibility Add out_dtype support for sparse semi-structured CUTLASS back-end (#116519) 2024-01-03 16:23:17 +00:00
functorch Impl. call_hasattr for BaseUserFunctionVariable (#116049) 2024-01-09 22:58:58 +00:00
fx [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
inductor Revert "[inductor] Add support for tl.make_block_ptr (#116079)" 2024-01-09 22:19:57 +00:00
jit Refactor some tests by using TEST_CUDA & TEST_MULTIGPU instead (#116083) 2024-01-03 08:53:59 +00:00
jit_hooks
lazy [BE]: Enable F821 and fix bugs (#116579) 2024-01-01 08:40:46 +00:00
mobile [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
nn [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
onnx [CPU] Add flash attention mask version (#115913) 2024-01-07 04:58:23 +00:00
onnx_caffe2
optim [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
package
profiler Add basic autograd TORCH_LOGS support (#115438) 2023-12-20 15:23:44 +00:00
quantization [quant][pt2e][xnnpack_quantizer] add support for linear_relu (#117052) 2024-01-09 23:19:52 +00:00
scripts
test_img
torch_np [dynamo] Fix np.issubdtype (#116459) 2024-01-05 01:48:07 +00:00
typing Add regression test for cuda_stream type checks (#111430) 2023-10-18 07:24:01 +00:00
_test_bazel.py
allowlist_for_publicAPI.json Remove dead tensor_has_hints. (#114071) 2023-11-20 16:02:24 +00:00
conftest.py Reset stepcurrent cache if file succeeds (#115775) 2023-12-19 22:19:57 +00:00
create_dummy_torchscript_model.py
delete.py
HowToWriteTestsUsingFileCheck.md
linear.py
load_torchscript_model.py
minioptest_failures_dict.json
mkl_verbose.py
mkldnn_verbose.py
run_doctests.sh
run_test.py [ROCm] Enabling additional UTs on ROCm (#115738) 2024-01-09 08:36:07 +00:00
simulate_nccl_errors.py
test_ao_sparsity.py
test_autocast.py Revert "More markDynamoStrictTest (#115870)" 2023-12-19 15:40:57 +00:00
test_autograd.py Make not passing use_reentrant back to warning instead of erroring and clarify docs (#116710) 2024-01-09 20:58:49 +00:00
test_autograd_fallback.py
test_binary_ufuncs.py [BE]: Apply FURB118 (prev): replaces unnecessary lambdas with operator. (#116027) 2023-12-20 19:35:08 +00:00
test_bundled_images.py [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
test_bundled_inputs.py
test_comparison_utils.py
test_compile_benchmark_util.py
test_complex.py
test_content_store.py Fix recompilation issue with content store (#113533) 2023-11-13 23:58:13 +00:00
test_cpp_api_parity.py markDynamoStrictTest some test_cpp_extensions.* (#115858) 2023-12-15 01:22:38 +00:00
test_cpp_extensions_aot.py markDynamoStrictTest some test_cpp_extensions.* (#115858) 2023-12-15 01:22:38 +00:00
test_cpp_extensions_jit.py markDynamoStrictTest some test_cpp_extensions.* (#115858) 2023-12-15 01:22:38 +00:00
test_cpp_extensions_open_device_registration.py markDynamoStrictTest some test_cpp_extensions.* (#115858) 2023-12-15 01:22:38 +00:00
test_cuda.py [CI] Test that cuInit is not called during import (#117010) 2024-01-09 14:44:22 +00:00
test_cuda_expandable_segments.py
test_cuda_multigpu.py
test_cuda_nvml_based_avail.py markDynamoStrictTest some more tests (#115857) 2023-12-15 01:22:38 +00:00
test_cuda_primary_ctx.py markDynamoStrictTest some more tests (#115857) 2023-12-15 01:22:38 +00:00
test_cuda_sanitizer.py
test_cuda_trace.py markDynamoStrictTest some more tests (#115857) 2023-12-15 01:22:38 +00:00
test_custom_ops.py Add config toggle suggestions for data-dependent/dynamic output shape (#114337) 2024-01-05 14:01:01 +00:00
test_dataloader.py [BE]: Enable RUFF PERF402 and apply fixes (#115505) 2023-12-20 18:01:24 +00:00
test_datapipe.py [BE]: Apply FURB118 (prev): replaces unnecessary lambdas with operator. (#116027) 2023-12-20 19:35:08 +00:00
test_decomp.py unMarkDynamoStrictTest on OpInfo-based tests (#115856) 2023-12-15 01:22:31 +00:00
test_deploy.py
test_determination.py
test_dispatch.py Revert "markDynamoStrictTest more tests (#115871)" 2023-12-19 15:36:27 +00:00
test_dlpack.py
test_dynamic_shapes.py add support for dynamic shapes in round (#115259) 2023-12-19 15:45:50 +00:00
test_expanded_weights.py [BE]: Enable F821 and fix bugs (#116579) 2024-01-01 08:40:46 +00:00
test_fake_tensor.py Add support for torch.cond in vmap (#114523) 2023-12-20 19:54:38 +00:00
test_flop_counter.py Re-add initial Flash Attention support on ROCM (#115981) 2024-01-04 22:21:31 +00:00
test_foreach.py Refactor some tests by using TEST_CUDA & TEST_MULTIGPU instead (#116083) 2024-01-03 08:53:59 +00:00
test_function_schema.py
test_functional_autograd_benchmark.py
test_functional_optim.py
test_functionalization.py Enable reverse view_funcs by default for python subclasses (#116512) 2024-01-05 16:48:12 +00:00
test_functionalization_of_rng_ops.py
test_futures.py
test_fx.py [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
test_fx_experimental.py
test_fx_passes.py
test_fx_reinplace_pass.py
test_hub.py
test_import_stats.py
test_indexing.py [BE]: Apply FURB118 (prev): replaces unnecessary lambdas with operator. (#116027) 2023-12-20 19:35:08 +00:00
test_itt.py
test_jit.py [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
test_jit_autocast.py
test_jit_disabled.py
test_jit_fuser.py [BE]: Enable F821 and fix bugs (#116579) 2024-01-01 08:40:46 +00:00
test_jit_fuser_legacy.py
test_jit_fuser_te.py [BE]: Enable F821 and fix bugs (#116579) 2024-01-01 08:40:46 +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 [BE]: Use iterable.chain.from_iterable where possible (#116376) 2023-12-27 19:20:07 +00:00
test_logging.py
test_masked.py
test_maskedtensor.py
test_matmul_cuda.py Skip unit tests that fail on MI210 runners (#114613) 2023-11-27 22:25:35 +00:00
test_meta.py [dynamo] Fix handling of one_hot (#116338) 2023-12-24 04:55:35 +00:00
test_metal.py
test_mkl_verbose.py
test_mkldnn.py [BE]: Enable RUFF PERF402 and apply fixes (#115505) 2023-12-20 18:01:24 +00:00
test_mkldnn_fusion.py improve mkldnn_linear_pointwise performance for contiguous tensor with non default contiguous strides (#114939) 2023-12-01 23:30:07 +00:00
test_mkldnn_verbose.py
test_mobile_optimizer.py Revert "markDynamoStrictTest on more tests (#115879)" 2023-12-15 20:00:09 +00:00
test_model_dump.py
test_model_exports_to_core_aten.py Build and test ExecuTorch on PyTorch (#113364) 2023-11-16 02:19:58 +00:00
test_module_init.py Revert "markDynamoStrictTest on more tests (#115879)" 2023-12-15 20:00:09 +00:00
test_modules.py Revert "markDynamoStrictTest on more tests (#115879)" 2023-12-15 20:00:09 +00:00
test_monitor.py
test_mps.py [MPS] Add support for 64-bit index operations (#116942) 2024-01-09 16:56:49 +00:00
test_multiprocessing.py Revert "markDynamoStrictTest on more tests (#115879)" 2023-12-15 20:00:09 +00:00
test_multiprocessing_spawn.py Revert "markDynamoStrictTest on more tests (#115879)" 2023-12-15 20:00:09 +00:00
test_namedtensor.py Add some checks about Device and Layout when create/convert named tensor (#113628) 2023-11-14 20:40:27 +00:00
test_namedtuple_return_api.py [CPU] Add flash attention mask version (#115913) 2024-01-07 04:58:23 +00:00
test_native_functions.py
test_native_mha.py Re-add initial Flash Attention support on ROCM (#115981) 2024-01-04 22:21:31 +00:00
test_nestedtensor.py Support squeeze.dim for jagged NT (#116891) 2024-01-06 01:00:53 +00:00
test_nn.py Re-add initial Flash Attention support on ROCM (#115981) 2024-01-04 22:21:31 +00:00
test_nnapi.py
test_numba_integration.py
test_numpy_interop.py Fix NULL dereference in binary CPU ops (#115183) 2023-12-06 03:37:47 +00:00
test_openmp.py
test_ops.py Introduce reverse view_funcs (#115894) 2024-01-05 16:48:12 +00:00
test_ops_fwd_gradients.py unMarkDynamoStrictTest on OpInfo-based tests (#115856) 2023-12-15 01:22:31 +00:00
test_ops_gradients.py unMarkDynamoStrictTest on OpInfo-based tests (#115856) 2023-12-15 01:22:31 +00:00
test_ops_jit.py unMarkDynamoStrictTest on OpInfo-based tests (#115856) 2023-12-15 01:22:31 +00:00
test_optim.py Migrate repr, deterministic state_dict test to OptimizerInfo (#116496) 2023-12-28 19:49:04 +00:00
test_out_dtype_op.py
test_overrides.py [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
test_package.py
test_per_overload_api.py
test_prims.py Revert "markDynamoStrictTest some more (#115885)" 2023-12-15 19:51:24 +00:00
test_proxy_tensor.py Don't decompose functional ops in predispatch functionalization (#116383) 2023-12-28 11:54:04 +00:00
test_pruning_op.py [dynamo] Eagerly install guards (#111415) 2023-11-07 19:55:19 +00:00
test_public_bindings.py [DeviceMesh] Rename _device_mesh.py to device_mesh.py to prepare for beta (#115099) (#115193) 2023-12-08 08:44:32 +00:00
test_python_dispatch.py [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
test_pytree.py [pytree] extend pytree operations with is_leaf prediction function (#116419) 2024-01-09 19:50:08 +00:00
test_quantization.py enable cat for cuda bits types (#115044) 2023-12-06 00:05:18 +00:00
test_reductions.py [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
test_scatter_gather_ops.py
test_schema_check.py Revert "markDynamoStrictTest some more (#115885)" 2023-12-15 19:51:24 +00:00
test_segment_reductions.py
test_serialization.py Support pathlib.Path as input to torch.load when mmap=True (#116104) 2023-12-28 22:54:11 +00:00
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py
test_show_pickle.py Revert "markDynamoStrictTest some more (#115885)" 2023-12-15 19:51:24 +00:00
test_sort_and_select.py [BE]: Enable F821 and fix bugs (#116579) 2024-01-01 08:40:46 +00:00
test_sparse.py Add batched sparse CSR/CSC/BSR/BSC to sparse COO conversion support (#116206) 2024-01-07 19:42:02 +00:00
test_sparse_csr.py [ROCm] Enabling additional UTs on ROCm (#115738) 2024-01-09 08:36:07 +00:00
test_sparse_semi_structured.py Add out_dtype support for sparse semi-structured CUTLASS back-end (#116519) 2024-01-03 16:23:17 +00:00
test_spectral_ops.py [CUDA][cuFFT] Initialize CUDA context for cuFFT before execute is called (#110326) 2023-10-13 18:02:25 +00:00
test_stateless.py
test_static_runtime.py Skip test_fork_wait_4 and test_fork_wait_4_async (#112743) 2023-11-02 20:46:29 +00:00
test_subclass.py
test_sympy_utils.py Run sympy expressions with Python values / FX tracing (#113978) 2023-11-20 21:25:11 +00:00
test_tensor_creation_ops.py Fix NaN bug in torch.signal.windows.kaiser (#116470) 2024-01-08 22:24:52 +00:00
test_tensorboard.py
test_tensorexpr.py
test_tensorexpr_pybind.py
test_testing.py Back out "[Kineto] Initialize libkineto profilers during torch init process during pybind set-up (#112623)" (#116201) 2023-12-21 16:32:19 +00:00
test_throughput_benchmark.py
test_torch.py [BE]: Update flake8 to v6.1.0 and fix lints (#116591) 2024-01-03 06:04:44 +00:00
test_transformers.py [CPU] Add flash attention mask version (#115913) 2024-01-07 04:58:23 +00:00
test_type_hints.py
test_type_info.py
test_type_promotion.py [BE]: Apply FURB118 (prev): replaces unnecessary lambdas with operator. (#116027) 2023-12-20 19:35:08 +00:00
test_typing.py [BE] Revive test_typing (#111428) 2023-10-18 02:19:49 +00:00
test_unary_ufuncs.py Skip some flaky Dynamo tests (#116165) 2023-12-20 05:05:02 +00:00
test_utils.py Some tiny modification about torch.set/get_default_device (#116014) 2023-12-19 05:08:06 +00:00
test_view_ops.py
test_vulkan.py
test_weak.py [BE]: Enable RUF015 codebase wide (#115507) 2023-12-11 15:51:01 +00:00
test_xnnpack_integration.py