From: Roy Li Date: Fri, 14 Dec 2018 03:33:37 +0000 (-0800) Subject: Fix numpy conversion for int8 tensor X-Git-Tag: accepted/tizen/6.5/unified/20211028.231830~2248 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=0b9b965c1a8e087a07f0fb712791b824265670a4;p=platform%2Fupstream%2Fpytorch.git 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 --- 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; } }