| 419 | |
| 420 | |
| 421 | class TensorDescription: |
| 422 | def __init__(self, element, layout, alignment=1, complex_transform=ComplexTransform.none): |
| 423 | self.element = element |
| 424 | self.layout = layout |
| 425 | if element != DataType.void: |
| 426 | self.alignment = min(128 // DataTypeSize[self.element], alignment) |
| 427 | else: |
| 428 | self.alignment = alignment |
| 429 | self.complex_transform = complex_transform |
| 430 | |
| 431 | |
| 432 | def CalculateSmemUsagePerStage(operation): |