Commit 3d36696b authored by zhaochao's avatar zhaochao
Browse files

[DCU] fix bug with cannot import name 'use_lightop_w8a8' from 'transformer_engine.pytorch.utils'


Signed-off-by: default avatarzhaochao <zhaochao1@sugon.com>
parent 2fc4b10c
......@@ -6,7 +6,6 @@ import pytest
import torch
import transformer_engine as te
import transformer_engine_torch as tex
from transformer_engine.pytorch.utils import use_lightop_w8a8
from transformer_engine.pytorch.constants import TE_DType
from transformer_engine.pytorch.fp8 import (FP8GlobalStateManager, blockwise_fp8_block_len, int8_simulation_fp8)
from transformer_engine.pytorch.tensor.float8_blockwise_tensor import (
......
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