diff --git a/S1/hli28146_#114/ModSwish_cuda.py b/S1/hli28146_#114/ModSwish_cuda.py index b4d5e503..5aac0789 100644 --- 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;