diff options
Diffstat (limited to 'xmake/modules')
| -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 |