Skip to content

GitLab

Commit 08f14085 authored by wenjh's avatar wenjh
Browse files

Fix missing import

parent 8fda607c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
transformer_engine/pytorch/triton/cross_entropy.py
View file @ 08f14085
...@@ -7,6 +7,7 @@ ...@@ -7,6 +7,7 @@
from typing import Union from typing import Union
from functools import reduce from functools import reduce
from operator import mul from operator import mul
from torch.utils.cpp_extension import IS_HIP_EXTENSION
import torch import torch
import torch.distributed as dist import torch.distributed as dist
......
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