wip

pytorch · May 15, 2024 · e24fed0 · e24fed0
1 parent 0d0e83c
commit e24fed0
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/fbgemm_gpu/experimental/gemm/CMakeLists.txt b/fbgemm_gpu/experimental/gemm/CMakeLists.txt
@@ -9,13 +9,13 @@
 ################################################################################
 
 set(experimental_triton_python_source_files
-    triton/__init__.py
-    triton/fp8_gemm.py)
+    triton_gemm/__init__.py
+    triton_gemm/fp8_gemm.py)
 
 
 ################################################################################
 # Install Python 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
@@ -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
@@ -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/..._gpu/experimental/gemm/triton/__init__.py → ...experimental/gemm/triton_gemm/__init__.py b/..._gpu/experimental/gemm/triton/__init__.py → ...experimental/gemm/triton_gemm/__init__.py
diff --git a/..._gpu/experimental/gemm/triton/fp8_gemm.py → ...experimental/gemm/triton_gemm/fp8_gemm.py b/..._gpu/experimental/gemm/triton/fp8_gemm.py → ...experimental/gemm/triton_gemm/fp8_gemm.py