fix undefined in device code error (NVIDIA#1880)

qingshui · Nov 6, 2024 · d656afb · d656afb
1 parent 32e3c38
commit d656afb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/include/cute/container/array_subbyte.hpp b/include/cute/container/array_subbyte.hpp
@@ -176,6 +176,7 @@ struct subbyte_reference
   }
 
   // Address
+  CUTE_HOST_DEVICE
   subbyte_iterator<T> operator&() const {
     return {ptr_, idx_};
   }