[cherry-pick] [FRONTEND] Remove dependency on torch triton-lang#7519 (triton-lang#7532)

cherry pick of triton-lang#7519

Co-authored-by: Thomas Raoux <[email protected]>

Author: atalman

python/triton/tools/tensor_descriptor.py

@@ -1,8 +1,6 @@
 from dataclasses import dataclass
 from typing import List, Any
 from triton._utils import validate_block_shape
-from torch._subclasses.fake_tensor import FakeTensor
-from torch._subclasses.functional_tensor import FunctionalTensor
 
 
 @dataclass
@@ -18,7 +16,9 @@ def __post_init__(self):
         assert len(self.block_shape) == rank, f"rank mismatch: {self}"
         assert rank > 0, "rank must not be zero"
         assert rank <= 5, "rank cannot be more than 5"
-        if not isinstance(self.base, (FakeTensor, FunctionalTensor)):
+        ty = type(self.base)
+        type_name = f"{ty.__module__}.{ty.__name__}"
+        if type_name not in ("torch.FakeTensor", "torch.FunctionalTensor"):
             assert self.base.data_ptr() % 16 == 0, "base must be 16-byte aligned"
         validate_block_shape(self.block_shape)
         elem_bytes = self.base.dtype.itemsize