diff --git a/js/web/docs/webgpu-operators.md b/js/web/docs/webgpu-operators.md index fe46165ffb..cf21fe8ed1 100644 --- a/js/web/docs/webgpu-operators.md +++ b/js/web/docs/webgpu-operators.md @@ -49,7 +49,7 @@ Do not modify directly.* | FusedConv | com.microsoft(1+) | | | Gather | ai.onnx(1-10,11-12,13+) | | | GatherElements | ai.onnx(11-12,13+) | | -| Gelu | com.microsoft(1+) | | +| Gelu | ai.onnx(20+); com.microsoft(1+) | | | Gemm | ai.onnx(7-8,9-10,11-12,13+) | | | GlobalAveragePool | ai.onnx(1+); com.ms.internal.nhwc(1+) | | | GlobalMaxPool | ai.onnx(1+); com.ms.internal.nhwc(1+) | | diff --git a/onnxruntime/core/providers/js/js_execution_provider.cc b/onnxruntime/core/providers/js/js_execution_provider.cc index e51b53686f..e289cba956 100644 --- a/onnxruntime/core/providers/js/js_execution_provider.cc +++ b/onnxruntime/core/providers/js/js_execution_provider.cc @@ -127,6 +127,7 @@ class ONNX_OPERATOR_VERSIONED_KERNEL_CLASS_NAME(kJsExecutionProvider, kOnnxDomai class ONNX_OPERATOR_VERSIONED_KERNEL_CLASS_NAME(kJsExecutionProvider, kOnnxDomain, 12, 12, Clip); class ONNX_OPERATOR_KERNEL_CLASS_NAME(kJsExecutionProvider, kOnnxDomain, 13, Clip); class ONNX_OPERATOR_KERNEL_CLASS_NAME(kJsExecutionProvider, kOnnxDomain, 6, Elu); +class ONNX_OPERATOR_KERNEL_CLASS_NAME(kJsExecutionProvider, kOnnxDomain, 20, Gelu); class ONNX_OPERATOR_VERSIONED_KERNEL_CLASS_NAME(kJsExecutionProvider, kOnnxDomain, 6, 12, Relu); class ONNX_OPERATOR_VERSIONED_KERNEL_CLASS_NAME(kJsExecutionProvider, kOnnxDomain, 13, 13, Relu); class ONNX_OPERATOR_KERNEL_CLASS_NAME(kJsExecutionProvider, kOnnxDomain, 14, Relu); @@ -441,6 +442,7 @@ std::unique_ptr RegisterKernels() { KERNEL_CREATE_INFO_VERSIONED(12, 12, Clip), KERNEL_CREATE_INFO(13, Clip), KERNEL_CREATE_INFO(6, Elu), + KERNEL_CREATE_INFO(20, Gelu), KERNEL_CREATE_INFO_VERSIONED(6, 12, Relu), KERNEL_CREATE_INFO_VERSIONED(13, 13, Relu), KERNEL_CREATE_INFO(14, Relu), diff --git a/onnxruntime/core/providers/js/operators/unary.cc b/onnxruntime/core/providers/js/operators/unary.cc index 9082527e3a..ef977161bc 100644 --- a/onnxruntime/core/providers/js/operators/unary.cc +++ b/onnxruntime/core/providers/js/operators/unary.cc @@ -151,6 +151,9 @@ ONNX_OPERATOR_KERNEL_EX(Clip, kOnnxDomain, 13, kJsExecutionProvider, JSEP_CLASS_IMPL_ATTRIBUTE_FLOAT_DEFAULT(Elu, Elu, alpha, 1.0) JSEP_ELEMENTWISE_KERNEL(Elu, 6, Elu) +JSEP_KERNEL_IMPL(Gelu, Gelu) +JSEP_ELEMENTWISE_KERNEL(Gelu, 20, Gelu) + JSEP_KERNEL_IMPL(Relu, Relu) JSEP_ELEMENTWISE_VERSIONED_KERNEL(Relu, 6, 12, Relu) JSEP_ELEMENTWISE_VERSIONED_KERNEL(Relu, 13, 13, Relu)