Commit 929f72ab authored by rocking's avatar rocking
Browse files

Fix build error

parent 0d55111e
...@@ -202,9 +202,9 @@ int main(int argc, char* argv[]) ...@@ -202,9 +202,9 @@ int main(int argc, char* argv[])
ck::tensor_operation::device::device_conv2d_fwd_instance:: ck::tensor_operation::device::device_conv2d_fwd_instance::
add_device_conv2d_fwd_xdl_c_shuffle_nhwc_kyxc_nhwk_f16_instances(conv_ptrs); add_device_conv2d_fwd_xdl_c_shuffle_nhwc_kyxc_nhwk_f16_instances(conv_ptrs);
} }
else if constexpr(ck::is_same_v<ck::remove_cv_t<InDataType>, bhalf_t> && else if constexpr(ck::is_same_v<ck::remove_cv_t<InDataType>, ck::bhalf_t> &&
ck::is_same_v<ck::remove_cv_t<WeiDataType>, bhalf_t> && ck::is_same_v<ck::remove_cv_t<WeiDataType>, ck::bhalf_t> &&
ck::is_same_v<ck::remove_cv_t<OutDataType>, bhalf_t>) ck::is_same_v<ck::remove_cv_t<OutDataType>, ck::bhalf_t>)
{ {
ck::tensor_operation::device::device_conv2d_fwd_instance:: ck::tensor_operation::device::device_conv2d_fwd_instance::
add_device_conv2d_fwd_xdl_nhwc_kyxc_nhwk_bf16_instances(conv_ptrs); add_device_conv2d_fwd_xdl_nhwc_kyxc_nhwk_bf16_instances(conv_ptrs);
...@@ -298,7 +298,7 @@ int main(int argc, char* argv[]) ...@@ -298,7 +298,7 @@ int main(int argc, char* argv[])
} }
else if(data_type == 2) else if(data_type == 2)
{ {
Run(bhalf_t(), bhalf_t(), bhalf_t()); Run(ck::bhalf_t(), ck::bhalf_t(), ck::bhalf_t());
} }
else if(data_type == 3) else if(data_type == 3)
{ {
......
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