From 0a0af30f4f7fb0470c6c2f13e5091525e450dba5 Mon Sep 17 00:00:00 2001
From: Zhenyi Zheng <57817774+ZDJeffrey@users.noreply.github.com>
Date: Thu, 15 May 2025 11:51:14 +0800
Subject: [PATCH] fix: remove redundant code to avoid error in specific env

---
 yunchang/ring/ring_flashinfer_attn.py | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/yunchang/ring/ring_flashinfer_attn.py b/yunchang/ring/ring_flashinfer_attn.py
index e8f8bed..e38f2c2 100644
--- a/yunchang/ring/ring_flashinfer_attn.py
+++ b/yunchang/ring/ring_flashinfer_attn.py
@@ -1,12 +1,8 @@
 import torch
 import torch.distributed as dist
 
-# from flash_attn.flash_attn_interface import _flash_attn_forward, _flash_attn_backward
 from .utils import RingComm, update_out_and_lse
 from yunchang.kernels import select_flash_attn_impl, AttnType
-import torch.utils.cpp_extension as torch_cpp_ext
-
-torch_cpp_ext._get_cuda_arch_flags()
 
 
 def ring_flashinfer_attn_forward(