diff --git a/mx_driving/csrc/SparseConv3d.cpp b/mx_driving/csrc/SparseConv3d.cpp
index e92532cf98875657398b4914d78242ffa85c78b8..4a6ee30f5a66c3832a75ca7f7b83b3304261b591 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;