saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-26 19:52:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix Python Packaging CI (Rocm) (#12477)

Browse source 
Fix Python Packaging CI

Co-authored-by: Ethan Tao <ettao@microsoft.com@orttrainingdev7.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
This commit is contained in:
ytaous 2022-08-04 20:40:09 -07:00 committed by GitHub
parent 8d830adf24
commit b879dca51c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/providers/rocm/miopen_common.cc
View file

@ -81,10 +81,12 @@ miopenDataType_t MiopenTensor::GetDataType() {
ORT_THROW("miopen engine currently supports only single/half/int32/int8 precision data types.");
}
#if ROCM_VERSION >= 50000
template<>
miopenDataType_t MiopenTensor::GetDataType<double>() {
return miopenDouble;
}
#endif
template<>
miopenDataType_t MiopenTensor::GetDataType<float>() {