MCPcopy Create free account
hub / github.com/KnowingNothing/MatmulTutorial / __init__

Method __init__

cutlass.py/tiling.py:92–106  ·  view source on GitHub ↗

dimensions: each item: if item is IntegerType, it's dynamic, else it's static int value.

(
        self, ndim: int, dimensions: Optional[List[Union[int, IntegerType]]] = None
    )

Source from the content-addressed store, hash-verified

90 """
91
92 def __init__(
93 self, ndim: int, dimensions: Optional[List[Union[int, IntegerType]]] = None
94 ) -> None:
95 """
96 dimensions:
97 each item: if item is IntegerType, it's dynamic, else it's static int value.
98 """
99 self.ndim = ndim
100 self.dimensions = (
101 ([IntegerType(unsigned=False, bits=32)] for _ in range(ndim))
102 if dimensions is None
103 else dimensions
104 )
105 assert isinstance(dimensions, (list, tuple))
106 assert len(self.dimensions) == self.ndim
107
108 def __getitem__(self, key: int):
109 assert key >= 0 and key < self.ndim

Callers

nothing calls this directly

Calls 1

IntegerTypeClass · 0.90

Tested by

no test coverage detected