False warning about unsupported doubles in runtime GEMM operations #2353
Labels
bug
An unexpected problem or unintended behavior
pending on user response
Waiting for more information or validation from user
Describe the bug
When writing a TF/keras model trained w/ with F64, tf2onnx warns about a lack of float64 support for GEMM by the runtime:
WARNING:root:For now, onnxruntime only support float32 type for Gemm rewriter
From:
https://github.com/onnx/tensorflow-onnx/blob/main/tf2onnx/rewriter/gemm_rewriter.py#L74
As far as I can tell ORT indeed does support double GEMM and was implemented about a year after this warning was added to the tf2onnx writer. I'm guessing that this warning was just never updated following the addition of double support. I've tested the exported F64 onnx models and they appear to be using doubles throughout the computation.
See this commit to the runtime repo adding double support:
microsoft/onnxruntime@5968a91
Based on this I believe we can remove that warning, or at least update it to represent a more current runtime support slate.
Urgency
LOW
To Reproduce
Export a model from keras trained in F64 using tf2onnx
The text was updated successfully, but these errors were encountered: