Unverified Commit 5210dc39 authored by Xudong Ma's avatar Xudong Ma Committed by GitHub
Browse files

[Misc] Update TritonLanguagePlaceholder to have attributes that are used by...


[Misc] Update TritonLanguagePlaceholder to have attributes that are used by Flash Linear Attention ops. (#26853)
Co-authored-by: default avatarXudong Ma <mxd@meta.com>
parent 650b51f9
...@@ -98,3 +98,6 @@ class TritonLanguagePlaceholder(types.ModuleType): ...@@ -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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment