add max(half, half) support when enable fp16

fix cpplint error, replace tab with whitespace
apache · Aug 21, 2019 · 2f8a10a · 2f8a10a
1 parent 929dff3
commit 2f8a10a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/codegen/codegen_cuda.cc b/src/codegen/codegen_cuda.cc
@@ -51,7 +51,7 @@ std::string CodeGenCUDA::Finish() {
   if (enable_fp16_) {
     decl_stream << "#include <cuda_fp16.h>\n";
     decl_stream << "__device__ half max(const half a, const half b)\n"
-                   "{\n\treturn __hgt(__half(a), __half(b)) ? a : b;\n}\n";
+                   "{\n  return __hgt(__half(a), __half(b)) ? a : b;\n}\n";
   }
 
   if (enable_int8_) {