From 81ce36e2d5e5eb0e6e06ae4eda96c7208b7f0c45 Mon Sep 17 00:00:00 2001
From: HDCharles <charlesdavidhernandez@gmail.com>
Date: Wed, 16 Oct 2024 15:04:52 -0700
Subject: [PATCH] adding BC for TensorCoreTiledLayoutType

Summary:

trying to land diff train, this is causing issues until torchtune can
correct their version guard, issue: https://github.com/pytorch/torchtune/issues/1855

Test Plan: see CI

Reviewers:

Subscribers:

Tasks:

Tags:
---
 torchao/dtypes/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/torchao/dtypes/__init__.py b/torchao/dtypes/__init__.py
index 4ab0c3f70..6e30bff95 100644
--- a/torchao/dtypes/__init__.py
+++ b/torchao/dtypes/__init__.py
@@ -17,6 +17,8 @@
     Float8AQTTensorImpl,
     MarlinSparseLayout,
 )
+# to maintain BC for torchtune, TODO remove this
+from .affine_quantized_tensor import TensorCoreTiledLayout as TensorCoreTiledLayoutType
 
 __all__ = [
     "NF4Tensor",