diff options
| author | Hoildkv <[email protected]> | 2022-02-25 20:24:35 +0800 |
|---|---|---|
| committer | GitHub <[email protected]> | 2022-02-25 20:24:35 +0800 |
| commit | 0722a44d0c64f78b019c7f439ed240e7d7bf3f96 (patch) | |
| tree | dd9ac892e56e8408686ad3ce01944453903408bc | |
| parent | 949795859315dbe358e3340967e35b09811fcbb8 (diff) | |
add plat/arch info to find_cudadevices
| -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 |