From 48322dbf27d98777b318ea7d4ef13ce084ebae1f Mon Sep 17 00:00:00 2001
From: zhangyue207 <zhangyue@qiyuanlab.com>
Date: Wed, 19 Jul 2023 10:06:47 +0800
Subject: [PATCH] Conv Kernel FP16

---
 src/kernels/cuda/conv_half.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/kernels/cuda/conv_half.cc b/src/kernels/cuda/conv_half.cc
index 8c85cf05..936e96ba 100644
--- a/src/kernels/cuda/conv_half.cc
+++ b/src/kernels/cuda/conv_half.cc
@@ -33,7 +33,7 @@ struct ConvCuDnnPerfRecordObj : public PerfRecordObj {
 
 using ConvCuDnnPerfRecord = Ref<ConvCuDnnPerfRecordObj>;
 
-class convCudnnHalf : public Kernel {
+class convCudnnFP16 : public Kernel {
 
     static constexpr int N_ALGO = 8;
     static constexpr int N_MODE = 2;
@@ -294,7 +294,7 @@ class convCudnnHalf : public Kernel {
     }
 };
 
-REGISTER_KERNEL(Device::CUDA, OpType::Conv, DataType::Float16, convCudnnHalf,
+REGISTER_KERNEL(Device::CUDA, OpType::Conv, DataType::Float16, convCudnnFP16,
                 "Conv_cuDNN_CUDA_Float16");
 
 REGISTER_CONSTRUCTOR(1, ConvCuDnnPerfRecordObj::from_json);