saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-11 00:49:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Support MLFloat16 type in Pow opset-12 CUDA kernel (#6233)

Browse source
This commit is contained in:
Hariharan Seshadri 2020-12-31 10:11:59 +05:30 committed by GitHub
parent 39a988ce1c
commit 4cc2ffef21
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
onnxruntime/core/providers/cuda/math/binary_elementwise_ops.cc
View file

@ -288,7 +288,7 @@ ONNX_OPERATOR_VERSIONED_KERNEL_EX(
kOnnxDomain,
12, 12,
kCudaExecutionProvider,
KernelDefBuilder().TypeConstraint("T", BuildKernelDefConstraints<int32_t, int64_t, float, double>()).TypeConstraint("T1", BuildKernelDefConstraints<int32_t, int64_t, float, double>()),
KernelDefBuilder().TypeConstraint("T", BuildKernelDefConstraints<int32_t, int64_t, float, double, MLFloat16>()).TypeConstraint("T1", BuildKernelDefConstraints<int32_t, int64_t, float, double, MLFloat16>()),
Pow);
ONNX_OPERATOR_KERNEL_EX(