Skip to content

Commit 7edc8a0

Browse files
committed
final diff
1 parent 42496d8 commit 7edc8a0

File tree

2 files changed

+8
-3
lines changed

2 files changed

+8
-3
lines changed

ext/ReactantCUDAExt.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,7 @@ function compile(job)
282282
entry = GPUCompiler.JuliaContext() do ctx
283283
mod, meta = GPUCompiler.compile(
284284
# :llvm, job; optimize=false, cleanup=false, validate=false, libraries=true
285-
:llvm, job; optimize=false, cleanup=false, validate=false, libraries=true
285+
:llvm, job; optimize=false, cleanup=false, validate=false, libraries=false
286286
# :llvm, job; optimize=false, cleanup=false, validate=true, libraries=false
287287
# :llvm, job; optimize=false, cleanup=false, validate=false, libraries=false
288288
)
@@ -323,7 +323,7 @@ function compile(job)
323323
end
324324
end
325325

326-
GPUCompiler.check_ir(job, mod)
326+
# GPUCompiler.check_ir(job, mod)
327327

328328
LLVM.strip_debuginfo!(mod)
329329
modstr = string(mod)

src/utils.jl

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,9 @@ function should_rewrite_ft(@nospecialize(ft))
116116
if ft.name.name == Symbol("#launch_configuration")
117117
return false
118118
end
119+
if ft.name.name == Symbol("cudaconvert")
120+
return false
121+
end
119122
end
120123
end
121124
end
@@ -161,7 +164,9 @@ function should_rewrite_ft(@nospecialize(ft))
161164
ft <: typeof(Base.getproperty) ||
162165
ft <: typeof(Base.vect) ||
163166
ft <: typeof(Base.eltype) ||
164-
ft <: typeof(Base.argtail)
167+
ft <: typeof(Base.argtail) ||
168+
ft <: typeof(Base.identity) ||
169+
ft <: typeof(Adapt.adapt_structure)
165170
return false
166171
end
167172

0 commit comments

Comments
 (0)