diff options
| author | OpportunityLiu <[email protected]> | 2025-12-01 12:47:59 +0800 |
|---|---|---|
| committer | OpportunityLiu <[email protected]> | 2025-12-01 12:47:59 +0800 |
| commit | 4846ee47d9f0bbe2cf05daa6e4fbbdd202cd86da (patch) | |
| tree | e9e7e1244ee4c525b38655ea900c5f50ac87ee26 /xmake/modules/lib | |
| parent | c7e1e6519762517d87e366a9ec7569aa49273f48 (diff) | |
feat: add support for cuda 11~13; update templates and examples
Diffstat (limited to 'xmake/modules/lib')
| -rw-r--r-- | xmake/modules/lib/detect/find_cudadevices.lua | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/xmake/modules/lib/detect/find_cudadevices.lua b/xmake/modules/lib/detect/find_cudadevices.lua index 0f459c711..a6ae606a6 100644 --- a/xmake/modules/lib/detect/find_cudadevices.lua +++ b/xmake/modules/lib/detect/find_cudadevices.lua @@ -228,6 +228,7 @@ end function _order_by_flops(devices) + -- See https://github.com/NVIDIA/cuda-samples/blob/master/Common/helper_cuda_drvapi.h#L100 local ngpu_arch_cores_per_sm = { [30] = 192 @@ -248,6 +249,11 @@ function _order_by_flops(devices) , [87] = 128 , [89] = 128 , [90] = 128 + , [100] = 128 + , [103] = 128 + , [110] = 128 + , [120] = 128 + , [121] = 128 } for _, dev in ipairs(devices) do |