Fix issue mentioned in microsoft#5911

xadupre · Nov 24, 2020 · 3a0cdc8 · 3a0cdc8
1 parent 048bda5
commit 3a0cdc8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/onnxruntime/core/providers/cpu/tensor/transpose.cc b/onnxruntime/core/providers/cpu/tensor/transpose.cc
@@ -3,7 +3,9 @@
 
 #include "core/providers/cpu/tensor/transpose.h"
 #include "core/framework/utils.h"
+#include "core/mlas/inc/mlas.h"
 #include "utils.h"
+
 namespace onnxruntime {
 
 /* A permutation [a,b,c,...] indicates that
@@ -135,7 +137,7 @@ static void TypedDoTransposeEltWise(int64_t num_axes, const std::vector<int64_t>
 
   const uint8_t* local_source = source;
   for (size_t i = 0; i < num_blocks; ++i) {
-    CopyPrim<uint64_t>(target, local_source);
+    CopyPrim<T>(target, local_source);
     IncrementIndexAndComputeOffset(mindex.data(), naxes, local_source);
     target += sizeof(T);
   }