def forward(ctx, features, filters, indice_pairs, indice_pair_num, num_activate_out, algo, timer: CUDAKernelTimer = CUDAKernelTimer(False)): ctx.save_for_backward(indice_pairs, indice_pair_num, features, filters) ctx.algo = algo ctx.timer = timer try: return ops.indice_conv(features, filters, indice_pairs, indice_pair_num, num_activate_out, False, algo=algo, timer=timer) except Exception as e: msg = "[Exception|indice_conv]" msg += f"feat={features.shape},w={filters.shape},pair={indice_pairs.shape}," msg += f"pairnum={indice_pair_num},act={num_activate_out},algo={algo}" print(msg, file=sys.stderr) spconv_save_debug_data((indice_pairs, indice_pair_num)) raise e
def forward(ctx, features, filters, indice_pairs, indice_pair_num, num_activate_out, algo, timer: CUDAKernelTimer = CUDAKernelTimer(False)): ctx.save_for_backward(indice_pairs, indice_pair_num, features, filters) ctx.algo = algo ctx.timer = timer return ops.indice_conv(features, filters, indice_pairs, indice_pair_num, num_activate_out, False, algo=algo, timer=timer)