From 4ab891999a04e2aeee2377fb7fae69f2d6f6749f Mon Sep 17 00:00:00 2001
From: ashari4 <70242157+ashari4@users.noreply.github.com>
Date: Thu, 6 Jan 2022 09:28:22 -0800
Subject: [PATCH] fix hardcoded type (#10205)

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

diff --git a/orttraining/orttraining/eager/ort_aten.h b/orttraining/orttraining/eager/ort_aten.h
index 58bdf626d3..bb235aa2ac 100644
--- a/orttraining/orttraining/eager/ort_aten.h
+++ b/orttraining/orttraining/eager/ort_aten.h
@@ -51,7 +51,7 @@ OrtValue create_ort_value(
     {1,},
     &ort_val);
   auto* ort_tensor = ort_val.GetMutable<onnxruntime::Tensor>();
-  CopyVectorToTensor<int64_t>(invoker, &val, 1, *ort_tensor);
+  CopyVectorToTensor<T>(invoker, &val, 1, *ort_tensor);
   return ort_val;
 }