[Misc] Update TritonLanguagePlaceholder to have attributes that are used by Flash Linear Attention ops. (#26853)

Co-authored-by: Xudong Ma <mxd@meta.com>
2025-10-15 01:37:49 -07:00
parent 650b51f9f9
commit 5210dc3940
1 changed files with 3 additions and 0 deletions
--- a/vllm/triton_utils/importing.py
+++ b/vllm/triton_utils/importing.py
@@ -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