wip: T10 batched forward (validation)

2026-06-16 00:19:26 +08:00
parent d2a585c5cb
commit 67687ec8fe
17 changed files with 959 additions and 150 deletions
--- a/crates/xtrain-cuda/build.rs
+++ b/crates/xtrain-cuda/build.rs
@@ -35,6 +35,7 @@ fn main() {
        .file("../../csrc/ops/nn.cu")
        .file("../../csrc/ops/model.cu")
        .file("../../csrc/ops/optim.cu")
+        .file("../../csrc/ops/attention.cu")
        .compile("xtrain_cuda_kernels");
 }