saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +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, kOnnxDomain,
12, 12, 12, 12,
kCudaExecutionProvider, 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); Pow);
ONNX_OPERATOR_KERNEL_EX( ONNX_OPERATOR_KERNEL_EX(