Commit ffe0ffe9 authored by rocking's avatar rocking
Browse files

Sync the naming with the develop

parent 963e4a71
...@@ -28,7 +28,7 @@ using F16_F16_Tuple = ck::Tuple<F16, F16>; ...@@ -28,7 +28,7 @@ using F16_F16_Tuple = ck::Tuple<F16, F16>;
using F32_Tuple = ck::Tuple<F32>; using F32_Tuple = ck::Tuple<F32>;
using INT32_Tuple = ck::Tuple<INT32>; using I32_Tuple = ck::Tuple<I32>;
// GEMM layout // GEMM layout
using Row = ck::tensor_layout::gemm::RowMajor; using Row = ck::tensor_layout::gemm::RowMajor;
......
...@@ -27,7 +27,7 @@ void add_device_conv2d_bias_perlayer_quantization_int8_instances( ...@@ -27,7 +27,7 @@ void add_device_conv2d_bias_perlayer_quantization_int8_instances(
GNHWK, GNHWK,
int8_t, int8_t,
int8_t, int8_t,
INT32_Tuple, I32_Tuple,
int8_t, int8_t,
PassThrough, PassThrough,
PassThrough, PassThrough,
...@@ -42,7 +42,7 @@ void add_device_conv2d_bias_relu_perlayer_quantization_int8_instances( ...@@ -42,7 +42,7 @@ void add_device_conv2d_bias_relu_perlayer_quantization_int8_instances(
GNHWK, GNHWK,
int8_t, int8_t,
int8_t, int8_t,
INT32_Tuple, I32_Tuple,
int8_t, int8_t,
PassThrough, PassThrough,
PassThrough, PassThrough,
...@@ -95,7 +95,7 @@ struct DeviceOperationInstanceFactory<ck::tensor_operation::device::DeviceGroupe ...@@ -95,7 +95,7 @@ struct DeviceOperationInstanceFactory<ck::tensor_operation::device::DeviceGroupe
is_same_v<OutLayout, GNHWK>) is_same_v<OutLayout, GNHWK>)
{ {
if constexpr(is_same_v<InDataType, int8_t> && is_same_v<WeiDataType, int8_t> && if constexpr(is_same_v<InDataType, int8_t> && is_same_v<WeiDataType, int8_t> &&
is_same_v<DsDataType, INT32_Tuple> && is_same_v<OutDataType, int8_t>) is_same_v<DsDataType, I32_Tuple> && is_same_v<OutDataType, int8_t>)
{ {
if constexpr(is_same_v<Activation, PassThrough>) if constexpr(is_same_v<Activation, PassThrough>)
add_device_conv2d_bias_perlayer_quantization_int8_instances(op_ptrs); add_device_conv2d_bias_perlayer_quantization_int8_instances(op_ptrs);
......
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