Remove __all__ exports from module

Removed __all__ exports from cutlass_int4_packed_layout.py

Author: jainapurva

torchao/dtypes/__init__.py

@@ -14,7 +14,6 @@
 )
 from .nf4tensor import NF4Tensor, to_nf4
 from .uintx import (
-    CutlassInt4PackedLayout,
     Int4CPULayout,
     Int4XPULayout,
     Int8DynamicActInt4WeightCPULayout,
@@ -29,6 +28,7 @@
     to_marlinqqq_quantized_intx,
 )
 from .uintx.block_sparse_layout import BlockSparseLayout
+from .uintx.cutlass_int4_packed_layout import CutlassInt4PackedLayout
 from .utils import (
     Layout,
     PlainLayout,

torchao/dtypes/uintx/cutlass_int4_packed_layout.py

@@ -8,27 +8,18 @@
 import warnings
 
 warnings.warn(
-    "Importing from torchao.dtypes.uintx.cutlass_int4_packed_layout is deprecated. "
+    "Importing from torchao.dtypes is deprecated. "
     "Please use 'from torchao.prototype.dtypes import CutlassInt4PackedLayout' instead. "
     "This import path will be removed in torchao v0.16.0.",
     DeprecationWarning,
     stacklevel=2,
 )
 
 from torchao.prototype.dtypes.uintx.cutlass_int4_packed_layout import (  # noqa: F401
-    CutlassInt4PackedLayout,
-    Int4PackedTensorImpl,
-    _linear_int4_act_int4_weight_cutlass_check,
-    _linear_int4_act_int4_weight_cutlass_impl,
-    _linear_int8_act_int4_weight_cutlass_check,
-    _linear_int8_act_int4_weight_cutlass_impl,
+    CutlassInt4PackedLayout,  # noqa: F401
+    Int4PackedTensorImpl,  # noqa: F401
+    _linear_int4_act_int4_weight_cutlass_check,  # noqa: F401
+    _linear_int4_act_int4_weight_cutlass_impl,  # noqa: F401
+    _linear_int8_act_int4_weight_cutlass_check,  # noqa: F401
+    _linear_int8_act_int4_weight_cutlass_impl,  # noqa: F401
 )
-
-__all__ = [
-    "CutlassInt4PackedLayout",
-    "Int4PackedTensorImpl",
-    "_linear_int4_act_int4_weight_cutlass_check",
-    "_linear_int4_act_int4_weight_cutlass_impl",
-    "_linear_int8_act_int4_weight_cutlass_check",
-    "_linear_int8_act_int4_weight_cutlass_impl",
-]