Skip to content

GitLab

"...git@developer.sourcefind.cn:kecinstone/2024-pra-vllm.git" did not exist on "ead94d9383644234fa9f153a950ab8cdc334c588"
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 @@
from typing import Union
from functools import reduce
from operator import mul
from torch.utils.cpp_extension import IS_HIP_EXTENSION
import torch
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