Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 364 Bytes
Newer Older
Russell Bryant's avatar
[Misc] Add SPDX-License-Identifier headers to python source files (#12628)  
Russell Bryant committed
1
2 
# SPDX-License-Identifier: Apache-2.0
Mengqing Cao's avatar
[Bugfix] Fix triton import with local TritonPlaceholder (#17446)  
Mengqing Cao committed
3
4 
from vllm.triton_utils.importing import (HAS_TRITON, TritonLanguagePlaceholder,
                                         TritonPlaceholder)
Thomas Parnell's avatar
[Bugfix] Add custom Triton cache manager to resolve MoE MP issue (#6140)  
Thomas Parnell committed
5
Mengqing Cao's avatar
[Bugfix] Fix triton import with local TritonPlaceholder (#17446)  
Mengqing Cao committed
6
7
8
9
10
11
12
13 
if HAS_TRITON:
    import triton
    import triton.language as tl
else:
    triton = TritonPlaceholder()
    tl = TritonLanguagePlaceholder()

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