__init__.py 543 Bytes
Newer Older
1
# SPDX-License-Identifier: Apache-2.0
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3

4
5
from vllm.triton_utils.importing import (HAS_TRITON, TritonLanguagePlaceholder,
                                         TritonPlaceholder)
6

7
8
9
if HAS_TRITON:
    import triton
    import triton.language as tl
10
    import triton.language.extra.libdevice as tldevice
11
12
13
else:
    triton = TritonPlaceholder()
    tl = TritonLanguagePlaceholder()
14
    tldevice = TritonLanguagePlaceholder()
15

16
__all__ = ["HAS_TRITON", "triton", "tl", "tldevice"]