Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "3b01d72a64d2c82e6f28acb2d74e2b58bdab3c48"
Unverified Commit 77a6c9d2 authored by Chaitanya Sri Krishna Lolla's avatar Chaitanya Sri Krishna Lolla Committed by GitHub
Browse files

Remove unused imports from rocm mla kernel. (#3963)

parent e3e0bc50
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 7 deletions
python/sglang/srt/layers/attention/triton_ops/rocm_mla_decode_rope.py
View file @ 77a6c9d2
...@@ -20,19 +20,12 @@ It supports page size = 1. ...@@ -20,19 +20,12 @@ It supports page size = 1.
# https://github.com/ModelTC/lightllm/blob/96353e868a840db4d103138caf15ed9dbea8c186/lightllm/models/deepseek2/triton_kernel/gqa_flash_decoding_stage1.py # https://github.com/ModelTC/lightllm/blob/96353e868a840db4d103138caf15ed9dbea8c186/lightllm/models/deepseek2/triton_kernel/gqa_flash_decoding_stage1.py
# https://github.com/ModelTC/lightllm/blob/96353e868a840db4d103138caf15ed9dbea8c186/lightllm/models/deepseek2/triton_kernel/gqa_flash_decoding_stage2.py # https://github.com/ModelTC/lightllm/blob/96353e868a840db4d103138caf15ed9dbea8c186/lightllm/models/deepseek2/triton_kernel/gqa_flash_decoding_stage2.py
import argparse
import logging
import sys
import pytest
import torch
import triton import triton
import triton.language as tl import triton.language as tl
from sglang.srt.layers.attention.triton_ops.decode_attention import ( from sglang.srt.layers.attention.triton_ops.decode_attention import (
_decode_softmax_reducev_fwd, _decode_softmax_reducev_fwd,
) )
from sglang.srt.layers.rotary_embedding import DeepseekScalingRotaryEmbedding
def is_hip(): def is_hip():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment