From f0bc35c4bab68345fc21038d54d415b3bdfe5bc8 Mon Sep 17 00:00:00 2001
From: "Tang, Cheng" <souptc@gmail.com>
Date: Tue, 12 Oct 2021 13:44:46 -0700
Subject: [PATCH] fix a hardcode type (#9337)

---
 orttraining/orttraining/eager/ort_aten.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/orttraining/orttraining/eager/ort_aten.cpp b/orttraining/orttraining/eager/ort_aten.cpp
index 8fbdc51086..79f33c41a2 100644
--- a/orttraining/orttraining/eager/ort_aten.cpp
+++ b/orttraining/orttraining/eager/ort_aten.cpp
@@ -181,7 +181,7 @@ at::Tensor empty__memory_format(
   auto& invoker = GetORTInvoker(*device_opt);
   CreateMLValue(
     invoker.GetCurrentExecutionProvider().GetAllocator(0, OrtMemTypeDefault),
-    ort_scalar_type_from_aten(at::kFloat),
+    ort_scalar_type_from_aten(*dtype_opt),
     size.vec(),
     &ot);