From 3a0cdc800389f103c59badc0c798c8a78999ce2b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?xavier=20dupr=C3=A9?= <xavier.dupre@gmail.com>
Date: Tue, 24 Nov 2020 20:42:10 +0100
Subject: [PATCH] Fix issue mentioned in #5911

---
 onnxruntime/core/providers/cpu/tensor/transpose.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/onnxruntime/core/providers/cpu/tensor/transpose.cc b/onnxruntime/core/providers/cpu/tensor/transpose.cc
index ba5d8de964bf4..fcd56de30d8c8 100644
--- 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);
   }