Skip to content

Commit 9aec916

Browse files
committed
fix: override to overlay
1 parent 57f8ab0 commit 9aec916

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ext/ReactantCUDAExt.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,7 @@ function transpose_val(val)
317317
return MLIR.IR.result(MLIR.Dialects.stablehlo.transpose(val; permutation=attr), 1)
318318
end
319319

320-
Reactant.@reactant_override @noinline function (func::LLVMFunc{F,tt})(args...; convert=Val(false), blocks::CuDim=1, threads::CuDim=1,
320+
Reactant.@reactant_overlay @noinline function (func::LLVMFunc{F,tt})(args...; convert=Val(false), blocks::CuDim=1, threads::CuDim=1,
321321
cooperative::Bool=false, shmem::Integer=0, call_kwargs...) where{F, tt}
322322
@show call_kwargs
323323

@@ -379,7 +379,7 @@ function compiler_cache(ctx::MLIR.IR.Context)
379379
return cache
380380
end
381381

382-
Reactant.@reactant_override @noinline function CUDA.cufunction(f::F, tt::TT=Tuple{}; kwargs...) where {F,TT}
382+
Reactant.@reactant_overlay @noinline function CUDA.cufunction(f::F, tt::TT=Tuple{}; kwargs...) where {F,TT}
383383
res = Base.@lock CUDA.cufunction_lock begin
384384
# compile the function
385385
cache = compiler_cache(MLIR.IR.context())

0 commit comments

Comments
 (0)