Skip to content

Commit 9653798

Browse files
committed
fix
1 parent 55813c7 commit 9653798

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
@@ -1455,8 +1455,8 @@ end
14551455

14561456
function __init__()
14571457
if CUDA.functional() && !Reactant.precompiling()
1458-
target = CUDA._compiler_config(CUDA.device()).target
1459-
Reactant.Compiler.cubinChip[] = "sm_$(target.cap.major)$(target.cap.minor)"
1458+
cap = CUDA.capability(CUDA.device())
1459+
Reactant.Compiler.cubinChip[] = "sm_$(cap.major)$(cap.minor)"
14601460
end
14611461
return nothing
14621462
end

0 commit comments

Comments
 (0)