ushrot change to ck::bhalf_t

device_operation/src/device_conv2d_bwd_data_xdl_nhwc_kyxc_nhwk_bf16_instance.cpp

@@ -9,7 +9,7 @@ namespace tensor_operation {
 namespace device {
 namespace device_conv2d_bwd_data_instance {
 
-using BF16 = ushort;
+using BF16 = ck::bhalf_t;
 using F32  = float;
 
 template <ck::index_t... Is>

profiler/include/profile_conv_bwd_data_impl.hpp

@@ -11,7 +11,7 @@
 
 using F16  = ck::half_t;
 using F32  = float;
-using BF16 = ushort;
+using BF16 = ck::bhalf_t;
 using INT8 = int8_t;
 namespace ck {
 namespace tensor_operation {
@@ -172,9 +172,9 @@ void profile_conv_bwd_data_impl(int do_verification,
         ck::tensor_operation::device::device_conv2d_bwd_data_instance::
             add_device_conv2d_bwd_data_xdl_nhwc_kyxc_nhwk_f16_instances(conv_ptrs);
     }
-    else if constexpr(ck::is_same_v<ck::remove_cv_t<InDataType>, ushort> &&
-                      ck::is_same_v<ck::remove_cv_t<WeiDataType>, ushort> &&
-                      ck::is_same_v<ck::remove_cv_t<OutDataType>, ushort>)
+    else if constexpr(ck::is_same_v<ck::remove_cv_t<InDataType>, ck::bhalf_t> &&
+                      ck::is_same_v<ck::remove_cv_t<WeiDataType>, ck::bhalf_t> &&
+                      ck::is_same_v<ck::remove_cv_t<OutDataType>, ck::bhalf_t>)
     {
         ck::tensor_operation::device::device_conv2d_bwd_data_instance::
             add_device_conv2d_bwd_data_xdl_nhwc_kyxc_nhwk_bf16_instances(conv_ptrs);

test/conv2d_bwd_data/conv2d_bwd_data.cpp

@@ -225,9 +225,9 @@ int main(int argc, char* argv[])
             ck::tensor_operation::device::device_conv2d_bwd_data_instance::
                 add_device_conv2d_bwd_data_xdl_nhwc_kyxc_nhwk_f16_instances(conv_ptrs);
         }
-        else if constexpr(ck::is_same_v<ck::remove_cv_t<InDataType>, ushort> &&
-                          ck::is_same_v<ck::remove_cv_t<WeiDataType>, ushort> &&
-                          ck::is_same_v<ck::remove_cv_t<OutDataType>, ushort>)
+        else if constexpr(ck::is_same_v<ck::remove_cv_t<InDataType>, ck::bhalf_t> &&
+                          ck::is_same_v<ck::remove_cv_t<WeiDataType>, ck::bhalf_t> &&
+                          ck::is_same_v<ck::remove_cv_t<OutDataType>, ck::bhalf_t>)
         {
             ck::tensor_operation::device::device_conv2d_bwd_data_instance::
                 add_device_conv2d_bwd_data_xdl_nhwc_kyxc_nhwk_bf16_instances(conv_ptrs);