CUTLASS 3.3.0 (#1167)
* Release 3.3.0 Adds support for mixed precision GEMMs On Hopper and Ampere Adds support for < 16B aligned GEMMs on Hopper Enhancements to EVT Enhancements to Python interface Enhancements to Sub-byte type handling in CuTe Several other bug-fixes and performance improvements. * minor doc update
This commit is contained in:
@ -34,7 +34,7 @@
|
||||
Utilities for expressing shapes
|
||||
"""
|
||||
|
||||
from cutlass import (
|
||||
from cutlass_library import (
|
||||
ConvMode,
|
||||
ConvKind,
|
||||
LayoutType
|
||||
@ -64,7 +64,7 @@ class MatrixCoord:
|
||||
Returns the leading dimension for a matrix with layout ``layout`` and shape provided by the MatrixCoord.
|
||||
|
||||
:param layout: layout of matrix
|
||||
:type layout: cutlass.LayoutType
|
||||
:type layout: cutlass_library.LayoutType
|
||||
|
||||
:returns: leading dimension
|
||||
:rtype: int
|
||||
|
||||
Reference in New Issue
Block a user