24 lines
640 B
Python
24 lines
640 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
from typing import TYPE_CHECKING
|
|
|
|
from vllm.triton_utils.importing import (
|
|
HAS_TRITON,
|
|
TritonLanguagePlaceholder,
|
|
TritonPlaceholder,
|
|
)
|
|
|
|
if TYPE_CHECKING or HAS_TRITON:
|
|
import triton
|
|
import triton.language as tl
|
|
import triton.language.extra.libdevice as tldevice
|
|
else:
|
|
triton = TritonPlaceholder()
|
|
tl = TritonLanguagePlaceholder()
|
|
tldevice = TritonLanguagePlaceholder()
|
|
|
|
LOG2E = 1.4426950408889634
|
|
LOGE2 = 0.6931471805599453
|
|
|
|
__all__ = ["HAS_TRITON", "triton", "tl", "tldevice", "LOG2E", "LOGE2"]
|