diff options
Diffstat (limited to 'xmake/rules')
| -rw-r--r-- | xmake/rules/cuda/gencodes/xmake.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/xmake/rules/cuda/gencodes/xmake.lua b/xmake/rules/cuda/gencodes/xmake.lua index f4f51c06d..f60f8cd2e 100644 --- a/xmake/rules/cuda/gencodes/xmake.lua +++ b/xmake/rules/cuda/gencodes/xmake.lua @@ -59,7 +59,7 @@ rule("cuda.gencodes") break end end - local device = find_cudadevices({skip_compute_mode_prohibited = true, order_by_flops = true, envs = cuda_envs})[1] + local device = find_cudadevices({skip_compute_mode_prohibited = true, order_by_flops = true, envs = cuda_envs, plat = target:plat(), arch = target:arch()})[1] if device then return nf_cugencode("sm_" .. device.major .. device.minor) end |