| commit | 88f1885ec9845a54c695463162f9d3fa1ae238bd | [log] [tgz] |
|---|---|---|
| author | maxren <maxren@meta.com> | Mon Jul 17 15:35:10 2023 -0700 |
| committer | PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> | Wed Jul 19 00:15:05 2023 +0000 |
| tree | 5c81779ceb29c2eb6e4b8e3ac267c759a6ad9cec | |
| parent | 39477f7ca977f6beac90d7c6708849f54f737a4f [diff] |
[XNNPACK][QS8] torch.cat (#104800) Differential Revision: [D47304143](https://our.internmc.facebook.com/intern/diff/D47304143/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/104800 Approved by: https://github.com/digantdesai
diff --git a/torch/ao/quantization/backend_config/executorch.py b/torch/ao/quantization/backend_config/executorch.py index 9546914..8c721fa 100644 --- a/torch/ao/quantization/backend_config/executorch.py +++ b/torch/ao/quantization/backend_config/executorch.py
@@ -416,6 +416,16 @@ .set_observation_type(ObservationType.OUTPUT_SHARE_OBSERVER_WITH_INPUT) .set_dtype_configs(dtype_configs) ) + cat_configs.append( + BackendPatternConfig(torch.concat) + .set_observation_type(ObservationType.OUTPUT_SHARE_OBSERVER_WITH_INPUT) + .set_dtype_configs(dtype_configs) + ) + cat_configs.append( + BackendPatternConfig(torch.concatenate) + .set_observation_type(ObservationType.OUTPUT_SHARE_OBSERVER_WITH_INPUT) + .set_dtype_configs(dtype_configs) + ) return cat_configs