From 9fa12e4de297d381b0b926f2bfef34b24b74f0ab Mon Sep 17 00:00:00 2001
From: Scott Roy <161522778+metascroy@users.noreply.github.com>
Date: Tue, 9 Sep 2025 10:35:57 -0700
Subject: [PATCH 1/2] Migrate tests from torchao/experimental

---
 ...chao_experimental_test.yml => regression_test_aarch64.yml} | 3 ++-
 .../prototype}/test_groupwise_lowbit_weight_lut_quantizer.py  | 4 ++++
 .../quantization}/test_embedding_xbit_quantizer.py            | 4 ++++
 3 files changed, 10 insertions(+), 1 deletion(-)
 rename .github/workflows/{torchao_experimental_test.yml => regression_test_aarch64.yml} (98%)
 rename {torchao/experimental/tests => test/prototype}/test_groupwise_lowbit_weight_lut_quantizer.py (96%)
 rename {torchao/experimental/tests => test/quantization}/test_embedding_xbit_quantizer.py (98%)

diff --git a/.github/workflows/torchao_experimental_test.yml b/.github/workflows/regression_test_aarch64.yml
similarity index 98%
rename from .github/workflows/torchao_experimental_test.yml
rename to .github/workflows/regression_test_aarch64.yml
index 575a80b0c3..eebe4b382e 100644
--- a/.github/workflows/torchao_experimental_test.yml
+++ b/.github/workflows/regression_test_aarch64.yml
@@ -1,4 +1,4 @@
-name: Run TorchAO Experimental Tests
+name: Run Regression Tests (aarch64)
 
 on:
   push:
@@ -44,6 +44,7 @@ jobs:
         if: runner.os == 'Linux'
         run: |
           conda activate venv
+          pip install coremltools
           pip install torch==2.7.0 --index-url https://download.pytorch.org/whl/cpu --force-reinstall
           pip install -r dev-requirements.txt
           BUILD_TORCHAO_EXPERIMENTAL=1 TORCHAO_BUILD_CPU_AARCH64=1 TORCHAO_BUILD_KLEIDIAI=1 TORCHAO_ENABLE_ARM_NEON_DOT=1 TORCHAO_PARALLEL_BACKEND=OPENMP pip install .
diff --git a/torchao/experimental/tests/test_groupwise_lowbit_weight_lut_quantizer.py b/test/prototype/test_groupwise_lowbit_weight_lut_quantizer.py
similarity index 96%
rename from torchao/experimental/tests/test_groupwise_lowbit_weight_lut_quantizer.py
rename to test/prototype/test_groupwise_lowbit_weight_lut_quantizer.py
index 1dae84b8a5..25d5398c50 100644
--- a/torchao/experimental/tests/test_groupwise_lowbit_weight_lut_quantizer.py
+++ b/test/prototype/test_groupwise_lowbit_weight_lut_quantizer.py
@@ -20,8 +20,12 @@
     group_size_to_block_shapes,
 )
 from torchao.quantization.quant_api import quantize_
+from torchao.quantization.quantize_.workflows.intx.intx_opaque_tensor import (
+    _is_kernel_library_loaded,
+)
 
 
+@unittest.skipIf(not _is_kernel_library_loaded(), "Need torchao lowbit kernels")
 class TestGroupwiseLowbitWeightLut(unittest.TestCase):
     """
     Test suite for the GroupwiseLutWeight quantization scheme, updated for the
diff --git a/torchao/experimental/tests/test_embedding_xbit_quantizer.py b/test/quantization/test_embedding_xbit_quantizer.py
similarity index 98%
rename from torchao/experimental/tests/test_embedding_xbit_quantizer.py
rename to test/quantization/test_embedding_xbit_quantizer.py
index 459c1c5e97..5b2b18e969 100644
--- a/torchao/experimental/tests/test_embedding_xbit_quantizer.py
+++ b/test/quantization/test_embedding_xbit_quantizer.py
@@ -32,9 +32,13 @@
     MappingType,
     quantize_,
 )
+from torchao.quantization.quantize_.workflows.intx.intx_opaque_tensor import (
+    _is_kernel_library_loaded,
+)
 from torchao.quantization.utils import compute_error
 
 
+@unittest.skipIf(not _is_kernel_library_loaded(), "Need torchao lowbit kernels")
 class TestEmbeddingQuantizer(unittest.TestCase):
     def test_accuracy(self):
         granularity = PerGroup(128)

From 1e7697fbb1ab3c85bf0bece2f8084b92d8e36970 Mon Sep 17 00:00:00 2001
From: Scott Roy <161522778+metascroy@users.noreply.github.com>
Date: Tue, 9 Sep 2025 10:37:24 -0700
Subject: [PATCH 2/2] up

---
 .github/workflows/regression_test_aarch64.yml | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/regression_test_aarch64.yml b/.github/workflows/regression_test_aarch64.yml
index eebe4b382e..d0a7eceead 100644
--- a/.github/workflows/regression_test_aarch64.yml
+++ b/.github/workflows/regression_test_aarch64.yml
@@ -51,11 +51,12 @@ jobs:
       - name: Run python tests
         run: |
           conda activate venv
-          pytest -s test/quantization/test_int8_dynamic_activation_intx_weight_config_v1.py
-          pytest -s torchao/experimental/tests/test_embedding_xbit_quantizer.py
           pytest -s torchao/experimental/tests/test_quant_passes.py
-          pytest -s test/prototype/test_dynamic_activation_lut.py
+          pytest -s test/quantization/test_int8_dynamic_activation_intx_weight_config_v1.py
+          pytest -s test/quantization/test_embedding_xbit_quantizer.py
           pytest -s test/quantization/quantize_/workflows/intx/test_intx_opaque_tensor.py
+          pytest -s test/prototype/test_dynamic_activation_lut.py
+          pytest -s test/prototype/test_groupwise_lowbit_weight_lut_quantizer.py
       - name: torchao/csrc/cpu - build and run C++ tests
         if: runner.os == 'macOS'
         run: |