From 0b9b965c1a8e087a07f0fb712791b824265670a4 Mon Sep 17 00:00:00 2001 From: Roy Li Date: Thu, 13 Dec 2018 19:33:37 -0800 Subject: [PATCH] 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 --- torch/csrc/utils/tensor_numpy.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/torch/csrc/utils/tensor_numpy.cpp b/torch/csrc/utils/tensor_numpy.cpp index 2700e51..5dfa57d 100644 --- 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; } } -- 2.7.4