diff --git a/backends/cuda/TARGETS b/backends/cuda/TARGETS
index 3892d5e1742..b1af2b0c352 100644
--- a/backends/cuda/TARGETS
+++ b/backends/cuda/TARGETS
@@ -55,6 +55,7 @@ runtime.python_library(
     srcs = [
         "triton/kernels/__init__.py",
         "triton/kernels/sdpa.py",
+        "triton/kernels/topk.py",
     ],
     visibility = [
         "//executorch/backends/cuda/...",