Commit d899c1e7 authored by wangshaojie6's avatar wangshaojie6
Browse files

change function name

parent 82b1b680
...@@ -1015,7 +1015,7 @@ struct DeviceConvndBwdWeightXdl_C_Shuffle_Input_N_Hi_Wi_C_Weight_K_Y_X_C_Output_ ...@@ -1015,7 +1015,7 @@ struct DeviceConvndBwdWeightXdl_C_Shuffle_Input_N_Hi_Wi_C_Weight_K_Y_X_C_Output_
}; };
// run kernel for bf16 with splitk // run kernel for bf16 with splitk
const auto Run_bf16_splitk = [&](const auto& kernel) { const auto run_bf16_splitk = [&](const auto& kernel) {
hipGetErrorString(hipMemset( hipGetErrorString(hipMemset(
arg.p_c_workspace_grid_, arg.p_c_workspace_grid_,
0, 0,
...@@ -1125,7 +1125,7 @@ struct DeviceConvndBwdWeightXdl_C_Shuffle_Input_N_Hi_Wi_C_Weight_K_Y_X_C_Output_ ...@@ -1125,7 +1125,7 @@ struct DeviceConvndBwdWeightXdl_C_Shuffle_Input_N_Hi_Wi_C_Weight_K_Y_X_C_Output_
remove_reference_t<DeviceOp::Block2CTileMap>, remove_reference_t<DeviceOp::Block2CTileMap>,
true>; true>;
Run_bf16_splitk(kernel_conv); run_bf16_splitk(kernel_conv);
ave_time += Run_type_convert(kernel_type_convert); ave_time += Run_type_convert(kernel_type_convert);
} }
} }
...@@ -1165,7 +1165,7 @@ struct DeviceConvndBwdWeightXdl_C_Shuffle_Input_N_Hi_Wi_C_Weight_K_Y_X_C_Output_ ...@@ -1165,7 +1165,7 @@ struct DeviceConvndBwdWeightXdl_C_Shuffle_Input_N_Hi_Wi_C_Weight_K_Y_X_C_Output_
remove_reference_t<DeviceOp::Block2CTileMap>, remove_reference_t<DeviceOp::Block2CTileMap>,
false>; false>;
Run_bf16_splitk(kernel); run_bf16_splitk(kernel);
} }
} }
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment