Merge branch 'ko3n1g/tests/fix-kitchen-guard' into 'core_r0.13.0'

tests: Fix kitchen guard

See merge request ADLR/megatron-lm!3515

Author: ko3n1g

.gitignore

@@ -11,4 +11,5 @@ local/
 .gitmodules
 wandb/
 onelogger.log
-onelogger.err
+onelogger.err
+.venv/

tests/unit_tests/models/test_gpt_model_quantization.py

@@ -16,13 +16,7 @@
     HAVE_TE = False
 
 try:
-    from megatron.core.extensions.kitchen import (
-        KitchenColumnParallelGroupedLinear,
-        KitchenColumnParallelLinear,
-        KitchenLayerNormColumnParallelLinear,
-        KitchenRowParallelGroupedLinear,
-        KitchenRowParallelLinear,
-    )
+    import nvidia_kitchen
 
     HAVE_KITCHEN = True
 except ImportError:
@@ -34,7 +28,6 @@
     not HAVE_TE, reason="Transformer Engine required for using kitchen backend with TE layers."
 )
 class TestGPTModelKitchenQuantizationConfig:
-
     def setup_method(self, method):
         Utils.initialize_model_parallel(1, 1)
         model_parallel_cuda_manual_seed(123)