From c210f2ea1d9595ae589c3342d409acab4a51e68b Mon Sep 17 00:00:00 2001 From: shidingfeng <251612113@qq.com> Date: Tue, 2 Jul 2024 13:01:18 +0800 Subject: [PATCH] fix illegal memory access error --- torchsparse/nn/functional/conv/kmap/build_kmap.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torchsparse/nn/functional/conv/kmap/build_kmap.py b/torchsparse/nn/functional/conv/kmap/build_kmap.py index 46d7e56..03ed0c6 100644 --- a/torchsparse/nn/functional/conv/kmap/build_kmap.py +++ b/torchsparse/nn/functional/conv/kmap/build_kmap.py @@ -190,6 +190,7 @@ def build_kernel_map( if dataflow == Dataflow.ImplicitGEMM: if training: + torch.cuda.synchronize() out_in_map_bwd = F.convert_transposed_out_in_map( kmap["out_in_map"], make_divisible(kmap["sizes"][0], cta_M) )