"vscode:/vscode.git/clone" did not exist on "293c7906e10c4baa41616b6a7b3d3b88ddad6e53"
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 ...@@ -6,7 +6,6 @@ import pytest
import torch import torch
import transformer_engine as te import transformer_engine as te
import transformer_engine_torch as tex 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.constants import TE_DType
from transformer_engine.pytorch.fp8 import (FP8GlobalStateManager, blockwise_fp8_block_len, int8_simulation_fp8) from transformer_engine.pytorch.fp8 import (FP8GlobalStateManager, blockwise_fp8_block_len, int8_simulation_fp8)
from transformer_engine.pytorch.tensor.float8_blockwise_tensor import ( 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