1

ykim362 · May 31, 2024 · fc45c3b · fc45c3b
1 parent 0ce98ba
commit fc45c3b
Showing 1 changed file with 0 additions and 20 deletions.
diff --git a/vllm/model_executor/layers/fused_moe/ampere_fp8_fused_moe.py b/vllm/model_executor/layers/fused_moe/ampere_fp8_fused_moe.py
@@ -4,8 +4,6 @@
 import json
 import os
 from typing import Any, Dict, Optional, Tuple, Callable
-import pkg_resources
-import shutil
 
 import torch
 import triton
@@ -18,24 +16,6 @@
 logger = init_logger(__name__)
 
 
-def replace_triton_cuda():
-
-    def get_package_path(package_name):
-        return pkg_resources.get_distribution(package_name).location
-
-    def get_package_version(package_name):
-        return pkg_resources.get_distribution(package_name).version
-
-    assert get_package_version('triton') == "2.2.0"
-
-    cur_folder_cuda_py = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'triton_cuda.py')
-    target_folder_cuda_py = os.path.join(get_package_path('triton'), 'triton', 'language', 'extra', 'cuda.py')
-    shutil.copyfile(cur_folder_cuda_py, target_folder_cuda_py)
-
-
-replace_triton_cuda()
-
-
 @triton.jit
 def fused_moe_kernel(
     # Pointers to matrices