commit 0f60c8d878ed45264ab56c5a9922dba00c7e4707
author Mikhail Zolotukhin <mvz@fb.com> Fri May 08 00:36:44 2020 -0700
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Fri May 08 00:40:47 2020 -0700
tree d14809e9a3a09bbedc96d8c9e5d3181fae553716
parent ff1a627bae80df4cc5018d96aa70c9efe09c6fe1

[TensorExpr] Correctly print 'bool' dtype in Cuda printer. (#38077)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/38077

Test Plan: Imported from OSS

Differential Revision: D21467298

Pulled By: ZolotukhinM

fbshipit-source-id: 65ac347f097e01aaf1d3ff5d598a402ca619d1f2

torch/csrc/jit/tensorexpr/cuda_codegen.cpp

diff --git a/torch/csrc/jit/tensorexpr/cuda_codegen.cpp b/torch/csrc/jit/tensorexpr/cuda_codegen.cpp
index 44d1593..298511e 100644
--- a/torch/csrc/jit/tensorexpr/cuda_codegen.cpp
+++ b/torch/csrc/jit/tensorexpr/cuda_codegen.cpp
@@ -107,6 +107,8 @@
 
 std::string cudaDtypeCppString(const Dtype& dtype) {
   switch (dtype.scalar_type()) {
+    case ScalarType::Bool:
+      return "bool";
     case ScalarType::Half:
       return "half";
     case ScalarType::Char:
@@ -117,9 +119,9 @@
       return "short";
     case ScalarType::Long:
       return "long";
-    default:; /* nothing */
+    default:
+      return dtype.ToCppString();
   }
-  return dtype.ToCppString();
 }
 
 static void print_flat_alloc(std::ostream& os, const Allocate* alloc) {