update

Author: bowang007

.github/workflows/build-test-linux-x86_64.yml

@@ -426,7 +426,9 @@ jobs:
         set -euo pipefail
         pushd .
         cd tests/py/dynamo
-        python -m pytest -ra -n 0 --junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_dynamo_plugins_tests_results.xml automatic_plugin/
+        python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_dynamo_plugins_tests_results.xml automatic_plugin/test_automatic_plugin.py
+        python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_dynamo_plugins_tests_results.xml automatic_plugin/test_automatic_plugin_with_attrs.py
+        python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_dynamo_plugins_tests_results.xml automatic_plugin/test_flashinfer_rmsnorm.py.py
         popd
 
   L2-torchscript-tests:

tests/py/dynamo/automatic_plugin/test_automatic_plugin.py

@@ -71,6 +71,7 @@ class TestAutomaticPlugin(DispatchTestCase):
     @parameterized.expand(
         [
             ((64, 64), torch.float),
+            ((256, 256), torch.int),
         ]
     )
     def test_mul_plugin_float(self, input_shape, dtype):

tests/py/dynamo/automatic_plugin/test_automatic_plugin_with_attrs.py

@@ -72,6 +72,7 @@ class TestAutomaticPlugin(DispatchTestCase):
     @parameterized.expand(
         [
             ((64, 64), torch.float),
+            ((256, 256), torch.int),
         ]
     )
     def test_scale_mul_plugin_float(self, input_shape, dtype):

tests/py/dynamo/automatic_plugin/test_flashinfer_rmsnorm.py

@@ -45,6 +45,7 @@ class TestAutomaticPlugin(DispatchTestCase):
     @parameterized.expand(
         [
             ((64, 64), (64,), torch.float16),
+            ((256, 256), (256,), torch.float16),
         ]
     )
     def test_rmsnorm_float(self, input_shape, weight_shape, data_type):