From e24fed0bbbadb87fb31c839a11c6738cdd56d72b Mon Sep 17 00:00:00 2001
From: Benson Ma <bensonma415@meta.com>
Date: Wed, 15 May 2024 14:51:14 -0700
Subject: [PATCH] wip

---
 fbgemm_gpu/experimental/gemm/CMakeLists.txt                 | 6 +++---
 fbgemm_gpu/experimental/gemm/test/fp8_gemm_benchmark.py     | 2 +-
 fbgemm_gpu/experimental/gemm/test/fp8_gemm_test.py          | 2 +-
 .../experimental/gemm/{triton => triton_gemm}/__init__.py   | 0
 .../experimental/gemm/{triton => triton_gemm}/fp8_gemm.py   | 0
 5 files changed, 5 insertions(+), 5 deletions(-)
 rename fbgemm_gpu/experimental/gemm/{triton => triton_gemm}/__init__.py (100%)
 rename fbgemm_gpu/experimental/gemm/{triton => triton_gemm}/fp8_gemm.py (100%)

diff --git a/fbgemm_gpu/experimental/gemm/CMakeLists.txt b/fbgemm_gpu/experimental/gemm/CMakeLists.txt
index 055b709e1a..ae0d236038 100644
--- a/fbgemm_gpu/experimental/gemm/CMakeLists.txt
+++ b/fbgemm_gpu/experimental/gemm/CMakeLists.txt
@@ -9,8 +9,8 @@
 ################################################################################
 
 set(experimental_triton_python_source_files
-    triton/__init__.py
-    triton/fp8_gemm.py)
+    triton_gemm/__init__.py
+    triton_gemm/fp8_gemm.py)
 
 
 ################################################################################
@@ -18,4 +18,4 @@ set(experimental_triton_python_source_files
 ################################################################################
 
 install(FILES ${experimental_triton_python_source_files}
-        DESTINATION fbgemm_gpu/experimental/gemm/triton)
+        DESTINATION fbgemm_gpu/experimental/gemm/triton_gemm)
diff --git a/fbgemm_gpu/experimental/gemm/test/fp8_gemm_benchmark.py b/fbgemm_gpu/experimental/gemm/test/fp8_gemm_benchmark.py
index 285d7fdad8..3ab1d36cf4 100644
--- a/fbgemm_gpu/experimental/gemm/test/fp8_gemm_benchmark.py
+++ b/fbgemm_gpu/experimental/gemm/test/fp8_gemm_benchmark.py
@@ -11,7 +11,7 @@
 import torch
 import triton
 
-from fbgemm_gpu.experimental.gemm.triton.fp8_gemm import (
+from fbgemm_gpu.experimental.gemm.triton_gemm.fp8_gemm import (
     matmul_fp8_block,
     matmul_fp8_row,
     quantize_fp8_block,
diff --git a/fbgemm_gpu/experimental/gemm/test/fp8_gemm_test.py b/fbgemm_gpu/experimental/gemm/test/fp8_gemm_test.py
index 642ac9d053..035ac3018d 100644
--- a/fbgemm_gpu/experimental/gemm/test/fp8_gemm_test.py
+++ b/fbgemm_gpu/experimental/gemm/test/fp8_gemm_test.py
@@ -11,7 +11,7 @@
 
 import torch
 
-from fbgemm_gpu.experimental.gemm.triton.fp8_gemm import (
+from fbgemm_gpu.experimental.gemm.triton_gemm.fp8_gemm import (
     matmul_fp8_block,
     matmul_fp8_row,
     quantize_fp8_block,
diff --git a/fbgemm_gpu/experimental/gemm/triton/__init__.py b/fbgemm_gpu/experimental/gemm/triton_gemm/__init__.py
similarity index 100%
rename from fbgemm_gpu/experimental/gemm/triton/__init__.py
rename to fbgemm_gpu/experimental/gemm/triton_gemm/__init__.py
diff --git a/fbgemm_gpu/experimental/gemm/triton/fp8_gemm.py b/fbgemm_gpu/experimental/gemm/triton_gemm/fp8_gemm.py
similarity index 100%
rename from fbgemm_gpu/experimental/gemm/triton/fp8_gemm.py
rename to fbgemm_gpu/experimental/gemm/triton_gemm/fp8_gemm.py