We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7fcb9df commit 3a3d991Copy full SHA for 3a3d991
src/Ops.jl
@@ -3225,4 +3225,18 @@ end
3225
)
3226
end
3227
3228
+@noinline function gelu(
3229
+ x::TracedRNumber{T},
3230
+ approximation::String;
3231
+ location=mlir_stacktrace("gelu", @__FILE__, @__LINE__),
3232
+) where {T}
3233
+ @assert approximation in ("NONE", "TANH", "SIGMOID")
3234
+ return TracedRNumber{T}(
3235
+ (),
3236
+ MLIR.IR.result(
3237
+ enzymexla.ml_gelu(x.mlir_data; gelu_approximation=approximation, location), 1
3238
+ ),
3239
+ )
3240
+end
3241
+
3242
end # module Ops
0 commit comments