2 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/include/nvgpu/bitops.h b/drivers/gpu/nvgpu/include/nvgpu/bitops.h
index 8760708e..b6666930 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/bitops.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/bitops.h
@@ -27,6 +27,8 @@
 /*
  * Explicit sizes for bit definitions. Please use these instead of BIT().
  */
+#define BIT8(i)         (U8(1)  << (i))
+#define BIT16(i)        (U16(1) << (i))
 #define BIT32(i)        (U32(1) << (i))
 #define BIT64(i)        (U64(1) << (i))
 
diff --git a/drivers/gpu/nvgpu/include/nvgpu/types.h b/drivers/gpu/nvgpu/include/nvgpu/types.h
index dfa95557..8320721b 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/types.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/types.h
@@ -55,6 +55,8 @@
  * overflows the 32 bit data type. So in this case we need an explicit cast to
  * 64 bits in order to prevent undefined behavior.
  */
+#define U8(x)   ((u8)(x))
+#define U16(x)  ((u16)(x))
 #define U32(x)  ((u32)(x))
 #define U64(x)  ((u64)(x))