saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-26 22:35:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix a hardcode type (#9337)

Browse source
This commit is contained in:
Tang, Cheng 2021-10-12 13:44:46 -07:00 committed by GitHub
parent d5c5c4fa50
commit f0bc35c4ba
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
orttraining/orttraining/eager/ort_aten.cpp
View file

@ -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);