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
......@@ -11,7 +11,6 @@ from typing import Any, Dict, List, Tuple, Union, Optional
import pytest
import torch
from pkg_resources import packaging
from transformer_engine.common import recipe
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