Skip to content

Commit 25836d0

Browse files
giordanovchuravy
andauthored
Allow for mistmatch between native target and params for IPU (#2552)
Co-authored-by: Mosè Giordano <[email protected]> Co-authored-by: Valentin Churavy <[email protected]>
1 parent f99b154 commit 25836d0

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/compiler.jl

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4292,7 +4292,10 @@ function GPUCompiler.compile_unhooked(output::Symbol, job::CompilerJob{<:EnzymeT
42924292
primal_target = (job.config.target::EnzymeTarget).target
42934293
primal_params = (job.config.params::EnzymeCompilerParams).params
42944294
if primal_target isa GPUCompiler.NativeCompilerTarget
4295-
@assert primal_params isa PrimalCompilerParams
4295+
if !(primal_params isa PrimalCompilerParams)
4296+
# XXX: This means mode is not propagated and rules are not applied for GPU code.
4297+
@safe_debug "NativeCompilerTarget without primal compiler params" primal_params
4298+
end
42964299
else
42974300
# XXX: This means mode is not propagated and rules are not applied for GPU code.
42984301
end

0 commit comments

Comments
 (0)