extract triton from triton_splitk

ghstack-source-id: c6675f5e50a7147e4c6f18a9ff2e64cecd795b18 Pull Request resolved: fairinternal/xformers#1239 __original_commit__ = fairinternal/xformers@9bed28e
facebookresearch · Oct 15, 2024 · c6c0ac3 · c6c0ac3
1 parent 916611d
commit c6c0ac3
Show file tree

Hide file tree

Showing 3 changed files with 1,151 additions and 1,119 deletions.
diff --git a/xformers/ops/fmha/_triton/__init__.py b/xformers/ops/fmha/_triton/__init__.py
@@ -0,0 +1,4 @@
+# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
+#
+# This source code is licensed under the BSD license found in the
+# LICENSE file in the root directory of this source tree.