From d4a68794000863900679abd133007830d41909b2 Mon Sep 17 00:00:00 2001
From: liyuke <liyuke@shanghaitech.edu.cn>
Date: Thu, 11 Sep 2025 07:17:09 +0000
Subject: [PATCH] test

---
 S1/ICVXKH/example_cudacode.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/S1/ICVXKH/example_cudacode.py b/S1/ICVXKH/example_cudacode.py
index 3232300..4e61f11 100644
--- a/S1/ICVXKH/example_cudacode.py
+++ b/S1/ICVXKH/example_cudacode.py
@@ -17,7 +17,7 @@ __global__ void relu_kernel(const float* x, float* y, int size) {
 torch::Tensor relu_cuda(torch::Tensor x) {
     auto size = x.numel();
     auto y = torch::empty_like(x);
-    const int block_size = 256;
+    const int block_size = 128;
     int num_blocks = (size + block_size - 1) / block_size;
     relu_kernel<<<num_blocks, block_size>>>(x.data_ptr<float>(), y.data_ptr<float>(), size);
     return y;
-- 
Gitee