Fix numpy conversion for int8 tensor

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

Differential Revision: D13459270

Pulled By: li-roy

fbshipit-source-id: 605534add263860a3ad9a7fa70888301ee0bf8e4

diff --git a/torch/csrc/utils/tensor_numpy.cpp b/torch/csrc/utils/tensor_numpy.cpp
index 2700e51..5dfa57d 100644 (file)
--- a/torch/csrc/utils/tensor_numpy.cpp
+++ b/torch/csrc/utils/tensor_numpy.cpp
@@ -153,6 +153,7 @@ static int aten_to_dtype(const at::Type& type) {
       case kInt: return NPY_INT32;
       case kShort: return NPY_INT16;
       case kByte: return NPY_UINT8;
+      case kChar: return NPY_INT8;
       default: break;
     }
   }