mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-10 04:26:00 +08:00
[Misc] Update TritonLanguagePlaceholder to have attributes that are used by Flash Linear Attention ops. (#26853)
Co-authored-by: Xudong Ma <mxd@meta.com>
This commit is contained in:
parent
650b51f9f9
commit
5210dc3940
@ -98,3 +98,6 @@ class TritonLanguagePlaceholder(types.ModuleType):
|
||||
self.int64 = None
|
||||
self.int32 = None
|
||||
self.tensor = None
|
||||
self.exp = None
|
||||
self.log = None
|
||||
self.log2 = None
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user