CUTLASS 3.6.0 (#1850)
* v3.6 * update changelog * update readme * fix typo * fixing typos * hopper gemm with weight prefetch --------- Co-authored-by: yuzhai <yuzhai@nvidia.com> Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
This commit is contained in:
co-authored by
yuzhai
Haicheng Wu
parent
0837a2a00a
commit
cc3c29a81a
@@ -94,9 +94,12 @@ using ${operation_name}_mainloop =
|
||||
${kernel_schedule}
|
||||
>::CollectiveOp;
|
||||
|
||||
using ${operation_name}_problem_shape = cutlass::conv::ConvProblemShape<${conv_kind}, ${operation_name}_mainloop::NumSpatialDimensions>;
|
||||
|
||||
// Unit tests call this "ConvKernel".
|
||||
// Conv operator ${operation_name}
|
||||
using ${operation_name}_base = cutlass::conv::kernel::ConvUniversal<
|
||||
${operation_name}_problem_shape,
|
||||
${operation_name}_mainloop,
|
||||
${operation_name}_epilogue,
|
||||
${tile_scheduler}
|
||||
|
||||
Reference in New Issue
Block a user