We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ddaa11f commit 3772dbdCopy full SHA for 3772dbd
python/triton/tools/tensor_descriptor.py
@@ -17,8 +17,7 @@ def __post_init__(self):
17
assert rank > 0, "rank must not be zero"
18
assert rank <= 5, "rank cannot be more than 5"
19
ty = type(self.base)
20
- type_name = f"{ty.__module__}.{ty.__name__}"
21
- if type_name not in ("torch.FakeTensor", "torch.FunctionalTensor"):
+ if ty.__name__ not in ("FakeTensor", "FunctionalTensor"):
22
assert self.base.data_ptr() % 16 == 0, "base must be 16-byte aligned"
23
validate_block_shape(self.block_shape)
24
elem_bytes = self.base.dtype.itemsize
0 commit comments