[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:
Xudong Ma
2025-10-15 01:37:49 -07:00
committed by GitHub
parent 650b51f9f9
commit 5210dc3940

View File

@ -98,3 +98,6 @@ class TritonLanguagePlaceholder(types.ModuleType):
self.int64 = None self.int64 = None
self.int32 = None self.int32 = None
self.tensor = None self.tensor = None
self.exp = None
self.log = None
self.log2 = None