Skip to content

Commit aa7c8ce

Browse files
authored
Add topk.py to BUCK
Differential Revision: D97649745 Pull Request resolved: pytorch#18399
1 parent f479ecf commit aa7c8ce

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

backends/cuda/TARGETS

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ runtime.python_library(
5555
srcs = [
5656
"triton/kernels/__init__.py",
5757
"triton/kernels/sdpa.py",
58+
"triton/kernels/topk.py",
5859
],
5960
visibility = [
6061
"//executorch/backends/cuda/...",

0 commit comments

Comments
 (0)