CUTLASS 3.4.0 (#1286)
* CUTLASS 3.4.0 * Update CHANGELOG.md --------- Co-authored-by: Pradeep Ramani <prramani@nvidia.com>
This commit is contained in:
@ -101,7 +101,7 @@ class Layout(LayoutBase):
|
||||
|
||||
# cosize(layout) Size of the codomain
|
||||
def cosize(self):
|
||||
return tuple_max(tuple((1, elem_scale(self.shape, self.stride))))
|
||||
return self(self.size() - 1) + 1
|
||||
|
||||
# print and str
|
||||
def __str__(self):
|
||||
|
||||
Reference in New Issue
Block a user