Skip to content

GitLab

Commit 2bfc08f1 authored by wangshaojie6's avatar wangshaojie6
Browse files

remove unused computation for work space size

parent 86449091
Show whitespace changes
Inline Side-by-side
Showing with 12 additions and 14 deletions
include/ck/tensor_operation/gpu/device/device_convnd_backward_weight_xdl_c_shuffle_nhwc_kyxc_nhwk.hpp
View file @ 2bfc08f1
...@@ -1180,14 +1180,13 @@ struct DeviceConvndBwdWeightXdl_C_Shuffle_Input_N_Hi_Wi_C_Weight_K_Y_X_C_Output_ ...@@ -1180,14 +1180,13 @@ struct DeviceConvndBwdWeightXdl_C_Shuffle_Input_N_Hi_Wi_C_Weight_K_Y_X_C_Output_
static size_t GetWorkSpaceSize(const Argument& arg) static size_t GetWorkSpaceSize(const Argument& arg)
{ {
size_t WorkSpaceSize = 0; size_t WorkSpaceSize = 0;
if(arg.k_batch_ > 1)
{
if constexpr(std::is_same<InDataType, ck::bhalf_t>::value) if constexpr(std::is_same<InDataType, ck::bhalf_t>::value)
{ {
WorkSpaceSize = WorkSpaceSize =
arg.Conv_K_ * arg.Conv_C_ * arg.filter_spatial_lengths_[0] * sizeof(float); arg.Conv_K_ * arg.Conv_C_ * arg.filter_spatial_lengths_[0] * sizeof(float);
} }
else
{
WorkSpaceSize = arg.Conv_K_ * 0;
} }
return WorkSpaceSize; return WorkSpaceSize;
} }
...@@ -1196,14 +1195,13 @@ struct DeviceConvndBwdWeightXdl_C_Shuffle_Input_N_Hi_Wi_C_Weight_K_Y_X_C_Output_ ...@@ -1196,14 +1195,13 @@ struct DeviceConvndBwdWeightXdl_C_Shuffle_Input_N_Hi_Wi_C_Weight_K_Y_X_C_Output_
static size_t GetWorkSpaceSize(const Argument& arg) static size_t GetWorkSpaceSize(const Argument& arg)
{ {
size_t WorkSpaceSize = 0; size_t WorkSpaceSize = 0;
if(arg.k_batch_ > 1)
{
if constexpr(std::is_same<InDataType, ck::bhalf_t>::value) if constexpr(std::is_same<InDataType, ck::bhalf_t>::value)
{ {
WorkSpaceSize = arg.Conv_K_ * arg.Conv_C_ * arg.filter_spatial_lengths_[0] * WorkSpaceSize = arg.Conv_K_ * arg.Conv_C_ * arg.filter_spatial_lengths_[0] *
arg.filter_spatial_lengths_[1] * sizeof(float); arg.filter_spatial_lengths_[1] * sizeof(float);
} }
else
{
WorkSpaceSize = arg.Conv_K_ * 0;
} }
return WorkSpaceSize; return WorkSpaceSize;
} }
......
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