Skip to content

GitLab

"git@developer.sourcefind.cn:kecinstone/2024-pra-vllm.git" did not exist on "ab3a5a8259922ce312d01be39d29e27666968039"
Unverified Commit 54c1cfad authored by Kirthi Shankar Sivamani's avatar Kirthi Shankar Sivamani Committed by GitHub
Browse files

[pytorch] removed unused import causing CI failures in fused attention (#1058) 



Rm unused import causing CI failures
Signed-off-by: default avatarKirthi Shankar Sivamani <ksivamani@nvidia.com>
parent d793ca17
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
tests/pytorch/fused_attn/test_fused_attn.py
View file @ 54c1cfad
...@@ -11,7 +11,6 @@ from typing import Any, Dict, List, Tuple, Union, Optional ...@@ -11,7 +11,6 @@ from typing import Any, Dict, List, Tuple, Union, Optional
import pytest import pytest
import torch import torch
from pkg_resources import packaging
from transformer_engine.common import recipe from transformer_engine.common import recipe
from transformer_engine.pytorch import TransformerLayer, fp8_autocast, fp8_model_init from transformer_engine.pytorch import TransformerLayer, fp8_autocast, fp8_model_init
......
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