From deed8b1256c93180acef24216b87f74f46f87681 Mon Sep 17 00:00:00 2001
From: Wesley Maxey <wesley.maxey@gmail.com>
Date: Thu, 21 Oct 2021 18:24:43 -0700
Subject: [PATCH] Add sm selectors for 87

---
 include/nv/target | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/include/nv/target b/include/nv/target
index 7675610e21..b6d75672fc 100644
--- a/include/nv/target
+++ b/include/nv/target
@@ -62,12 +62,13 @@ namespace nv {
       constexpr base_int_t sm_75_bit = 1 << 11;
       constexpr base_int_t sm_80_bit = 1 << 12;
       constexpr base_int_t sm_86_bit = 1 << 13;
+      constexpr base_int_t sm_87_bit = 1 << 13;
       constexpr base_int_t all_devices =
           sm_35_bit | sm_37_bit |
           sm_50_bit | sm_52_bit | sm_53_bit |
           sm_60_bit | sm_61_bit | sm_62_bit |
           sm_70_bit | sm_72_bit | sm_75_bit |
-          sm_80_bit | sm_86_bit;
+          sm_80_bit | sm_86_bit | sm_87_bit;
 
       // Store a set of targets as a set of bits
       struct _NV_BITSET_ATTRIBUTE target_description {
@@ -82,7 +83,7 @@ namespace nv {
         sm_50 = 50, sm_52 = 52, sm_53 = 53,
         sm_60 = 60, sm_61 = 61, sm_62 = 62,
         sm_70 = 70, sm_72 = 72, sm_75 = 75,
-        sm_80 = 80, sm_86 = 86,
+        sm_80 = 80, sm_86 = 86, sm_87 = 87,
       };
       _NV_FUNCTION_ANNOTATION
       constexpr base_int_t toint(sm_selector a) {
@@ -102,11 +103,13 @@ namespace nv {
                toint(a) == 72 ? sm_72_bit :
                toint(a) == 75 ? sm_75_bit :
                toint(a) == 80 ? sm_80_bit :
-               toint(a) == 86 ? sm_86_bit : 0;
+               toint(a) == 86 ? sm_86_bit :
+               toint(a) == 87 ? sm_87_bit : 0;
       }
       _NV_FUNCTION_ANNOTATION
       constexpr base_int_t bitrounddown(sm_selector a) {
-        return toint(a) >= 86 ? sm_86_bit :
+        return toint(a) >= 87 ? sm_87_bit :
+               toint(a) >= 86 ? sm_86_bit :
                toint(a) >= 80 ? sm_80_bit :
                toint(a) >= 75 ? sm_75_bit :
                toint(a) >= 72 ? sm_72_bit :
@@ -180,6 +183,7 @@ namespace nv {
     constexpr sm_selector sm_75 = sm_selector::sm_75;
     constexpr sm_selector sm_80 = sm_selector::sm_80;
     constexpr sm_selector sm_86 = sm_selector::sm_86;
+    constexpr sm_selector sm_87 = sm_selector::sm_87;
 
     using detail::is_exactly;
     using detail::provides;