unbreak gptq int4 test and add to CI (#4223)

Update

[ghstack-poisoned]

Author: vkuzo

.github/workflows/1xH100_tests.yml

@@ -60,3 +60,4 @@ jobs:
         pytest test/prototype/moe_training/test_fp8_grouped_mm.py --verbose -s
         pytest test/prototype/moe_training/test_mxfp8_grouped_mm.py --verbose -s
         pytest test/prototype/moe_training/test_training.py --verbose -s
+        pytest test/prototype/gptq/ --verbose -s

test/prototype/gptq/test_gptqv2.py

@@ -429,7 +429,7 @@ def test_gptq_quantize_function(self, base_config):
                 id="int4",
                 marks=pytest.mark.skipif(
                     not _is_mslk_available(),
-                    reason="fbgemm_gpu not available",
+                    reason="mslk not available",
                 ),
             ),
             pytest.param(

torchao/prototype/gptq/api.py

@@ -13,7 +13,7 @@
 import torch.nn as nn
 
 try:
-    from mslk.quantize import int4_row_quantize_zp, pack_int4
+    from mslk.quantize.shuffle import int4_row_quantize_zp, pack_int4
 except:
     int4_row_quantize_zp = None
     pack_int4 = None
@@ -71,7 +71,7 @@ def __post_init__(self):
         if isinstance(self.base_config, Int4WeightOnlyConfig):
             if int4_row_quantize_zp is None:
                 raise ValueError(
-                    "fbgemm_gpu is not installed. Please install fbgemm_gpu to use int4 quantization."
+                    "mslk is not installed. Please install mslk to use int4 quantization."
                 )