finish ModSwish #114

2025-12-10 19:27:42 +08:00 · 2025-12-10 19:27:42 +08:00 · 247a681616
parent 76d09aaf92
commit 247a681616
1 changed files with 1 additions and 1 deletions
--- a/S1/hli28146_#114/ModSwish_cuda.py
+++ b/S1/hli28146_#114/ModSwish_cuda.py
@ -65,7 +65,7 @@ torch::Tensor modswish_cuda_forward(const torch::Tensor& input) {
    TORCH_CHECK(input.is_contiguous(), "Input must be contiguous");
    
    const int n = input.numel();
-    auto output = torch.empty_like(input);
+    auto output = torch::empty_like(input);

    const int vec_n = n / 4;
    const int grid_size = (vec_n + BLOCK_SIZE - 1) / BLOCK_SIZE;