diff --git a/ext/ReactantCUDAExt.jl b/ext/ReactantCUDAExt.jl index 6b9e6360..414876e6 100644 --- a/ext/ReactantCUDAExt.jl +++ b/ext/ReactantCUDAExt.jl @@ -317,7 +317,7 @@ function transpose_val(val) return MLIR.IR.result(MLIR.Dialects.stablehlo.transpose(val; permutation=attr), 1) end -Reactant.@reactant_override @noinline function (func::LLVMFunc{F,tt})(args...; convert=Val(false), blocks::CuDim=1, threads::CuDim=1, +Reactant.@reactant_overlay @noinline function (func::LLVMFunc{F,tt})(args...; convert=Val(false), blocks::CuDim=1, threads::CuDim=1, cooperative::Bool=false, shmem::Integer=0, call_kwargs...) where{F, tt} @show call_kwargs