From 2024305b6bb7c17738b800267ba54a6990e0f918 Mon Sep 17 00:00:00 2001 From: Shan Zhou Date: Sun, 6 Feb 2022 08:02:46 -0800 Subject: Update xmake.lua Supports Ampere architecture. --- xmake/rules/cuda/gencodes/xmake.lua | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/xmake/rules/cuda/gencodes/xmake.lua b/xmake/rules/cuda/gencodes/xmake.lua index f83ce4740..f4f51c06d 100644 --- a/xmake/rules/cuda/gencodes/xmake.lua +++ b/xmake/rules/cuda/gencodes/xmake.lua @@ -43,8 +43,8 @@ rule("cuda.gencodes") -- sm_20 and compute_20 is supported until CUDA 8 -- sm_30 and compute_30 is supported until CUDA 10 - local known_v_archs = hashset.of(20, 30, 32, 35, 37, 50, 52, 53, 60, 61, 62, 70, 72, 75, 80) - local known_r_archs = hashset.of(20, 30, 32, 35, 37, 50, 52, 53, 60, 61, 62, 70, 72, 75, 80) + local known_v_archs = hashset.of(20, 30, 32, 35, 37, 50, 52, 53, 60, 61, 62, 70, 72, 75, 80, 86, 87) + local known_r_archs = hashset.of(20, 30, 32, 35, 37, 50, 52, 53, 60, 61, 62, 70, 72, 75, 80, 86, 87) local function nf_cugencode(archs) if type(archs) ~= "string" then -- cgit v1.3.1