From 24dd1a26b4de3a0b98ebe754d0e38a392efdffbf Mon Sep 17 00:00:00 2001
From: ZYF-Annarine <zhangyifan115@huawei.com>
Date: Tue, 24 Dec 2024 17:49:13 +0800
Subject: [PATCH] sparseconv3d outchannle support 256

---
 mx_driving/csrc/SparseConv3d.cpp | 2 --
 1 file changed, 2 deletions(-)

diff --git a/mx_driving/csrc/SparseConv3d.cpp b/mx_driving/csrc/SparseConv3d.cpp
index e92532cf..4a6ee30f 100644
--- a/mx_driving/csrc/SparseConv3d.cpp
+++ b/mx_driving/csrc/SparseConv3d.cpp
@@ -21,8 +21,6 @@ std::tuple<at::Tensor, at::Tensor> npu_sparse_conv3d(const at::Tensor& indices,
     at::IntArrayRef stride, at::IntArrayRef padding, int out_channel, at::IntArrayRef outSpatialShape, int batch_size)
 {
     TORCH_CHECK_NPU(indices);
-    TORCH_CHECK(
-        out_channel <= 128, "out_channel must less or equal than 128 expected but got out_channel: ", out_channel);
     TORCH_CHECK(out_channel % 8 == 0, "out_channel must be divisible by 8 but got out_channel: ", out_channel);
     auto indices_size = indices.sizes();
     int64_t kernelsum = 1;
-- 
Gitee