From 261fff9c48d6904d1f98ee3e864dcb857265de83 Mon Sep 17 00:00:00 2001
From: Colin Peppler <colinpeppler@meta.com>
Date: Thu, 27 Feb 2025 21:16:35 -0800
Subject: [PATCH] Specify uint8_t as enum size

---
 deep_gemm/include/deep_gemm/fp8_gemm.cuh  | 2 +-
 deep_gemm/include/deep_gemm/scheduler.cuh | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/deep_gemm/include/deep_gemm/fp8_gemm.cuh b/deep_gemm/include/deep_gemm/fp8_gemm.cuh
index 4a0c79e..6b8d4e6 100644
--- a/deep_gemm/include/deep_gemm/fp8_gemm.cuh
+++ b/deep_gemm/include/deep_gemm/fp8_gemm.cuh
@@ -16,7 +16,7 @@
 
 namespace deep_gemm {
 
-enum class Layout {
+enum class Layout : uint8_t {
     RowMajor,
     ColMajor
 };
diff --git a/deep_gemm/include/deep_gemm/scheduler.cuh b/deep_gemm/include/deep_gemm/scheduler.cuh
index 329fbb0..2cc37a8 100644
--- a/deep_gemm/include/deep_gemm/scheduler.cuh
+++ b/deep_gemm/include/deep_gemm/scheduler.cuh
@@ -2,7 +2,7 @@
 
 namespace deep_gemm {
 
-enum class GemmType {
+enum class GemmType : uint8_t {
     Normal,
     GroupedContiguous,
     GroupedMasked