Initial commit

pkgs/triton/ops/__init__.py

@@ -0,0 +1,17 @@
+# from .conv import _conv, conv
+from . import blocksparse
+from .cross_entropy import _cross_entropy, cross_entropy
+from .flash_attention import attention
+from .matmul import _matmul, matmul
+from .bmm_matmul import _bmm, bmm
+
+__all__ = [
+    "blocksparse",
+    "_cross_entropy",
+    "cross_entropy",
+    "_matmul",
+    "matmul",
+    "_bmm",
+    "bmm",
+    "attention",
+]