CUTLASS 3.1 (#915)

Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
This commit is contained in:
ANIKET SHIVAM
2023-04-14 23:19:34 -04:00
committed by GitHub
co-authored by Aniket Shivam
parent 9b8166e3f0
commit d572cc1aab
482 changed files with 37175 additions and 16410 deletions
@@ -651,7 +651,7 @@ Status Conv3dOperationProfiler::initialize_workspace(
if (options.execution_mode != ExecutionMode::kDryRun) {
int seed_shift = 0;
conv_workspace_.A = device_context.allocate_tensor(
options,
"A",
@@ -659,7 +659,8 @@ Status Conv3dOperationProfiler::initialize_workspace(
operation_desc.A.layout,
problem_.extent_a(operation_desc.conv_kind),
conv_workspace_.stride_a(operation_desc.conv_kind),
conv_workspace_.problem_count
conv_workspace_.problem_count,
seed_shift++
);
conv_workspace_.B = device_context.allocate_tensor(
@@ -669,7 +670,8 @@ Status Conv3dOperationProfiler::initialize_workspace(
operation_desc.B.layout,
problem_.extent_b(operation_desc.conv_kind),
conv_workspace_.stride_b(operation_desc.conv_kind),
conv_workspace_.problem_count
conv_workspace_.problem_count,
seed_shift++
);
conv_workspace_.C = device_context.allocate_tensor(
@@ -679,7 +681,8 @@ Status Conv3dOperationProfiler::initialize_workspace(
operation_desc.C.layout,
problem_.extent_c(operation_desc.conv_kind),
conv_workspace_.stride_c(operation_desc.conv_kind),
conv_workspace_.problem_count
conv_workspace_.problem_count,
seed_shift++
);
conv_workspace_.Computed = device_context.allocate_tensor(