From 9d411eaa8212e1b74cb16acc551792b598962f90 Mon Sep 17 00:00:00 2001 From: garryling Date: Mon, 1 Sep 2025 23:00:49 +0800 Subject: [PATCH] fix: split_k_iters == 32 is root case cause tensor large than 4GB, pointer offset overflow by using int32 range --- torchsparse/nn/functional/conv/func/implicit_gemm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchsparse/nn/functional/conv/func/implicit_gemm.py b/torchsparse/nn/functional/conv/func/implicit_gemm.py index e91ad18..c700541 100644 --- a/torchsparse/nn/functional/conv/func/implicit_gemm.py +++ b/torchsparse/nn/functional/conv/func/implicit_gemm.py @@ -161,7 +161,7 @@ def backward(ctx, grad_output: torch.Tensor): reorder_out_in_map_bwd, reduced_sorted_mask_bwd_wgrad, reorder_loc_bwd, - 32, + 1, torchsparse.backends.allow_tf32, torchsparse.backends.allow_fp16, torchsparse.backends.allow_bf16 @@ -192,7 +192,7 @@ def backward(ctx, grad_output: torch.Tensor): grad_output, input, out_in_map_bwd, - 32, + 1, torchsparse.backends.allow_tf32, torchsparse.backends.allow_fp16, torchsparse.backends.allow_bf16