CUTLASS 3.2.1 (#1113)
* Updates for 3.2.1 release. * Minor fix in gemm op profiler for raster order. * Add scheduler mapping for raster order in the kernels.
This commit is contained in:
@@ -1,6 +1,3 @@
|
||||
# module-wide variables
|
||||
import os
|
||||
|
||||
from cutlass.backend.arguments import *
|
||||
from cutlass.backend.c_types import *
|
||||
from cutlass.backend.compiler import ArtifactManager
|
||||
@@ -11,9 +8,7 @@ from cutlass.backend.gemm_operation import *
|
||||
from cutlass.backend.library import *
|
||||
from cutlass.backend.memory_manager import PoolMemoryManager
|
||||
from cutlass.backend.operation import *
|
||||
from cutlass.backend.parser import *
|
||||
from cutlass.backend.reduction_operation import *
|
||||
from cutlass.backend.tensor_ref import *
|
||||
from cutlass.backend.type_hint import *
|
||||
from cutlass.backend.utils import *
|
||||
from cutlass.backend.utils.device import device_cc
|
||||
|
||||
Reference in New Issue
Block a user