Upload spconv_stub/spconv/pytorch.py with huggingface_hub

spconv_stub/spconv/pytorch.py

@@ -0,0 +1,27 @@
+"""spconv.pytorch stub."""
+import torch
+import torch.nn as nn
+from enum import IntEnum
+
+class ConvAlgo:
+    Native = 0
+    MaskImplicitGemm = 1
+
+class SparseConvTensor:
+    def __init__(self, features=None, indices=None, spatial_shape=None, batch_size=1):
+        self.features = features
+        self.indices = indices
+        self.spatial_shape = spatial_shape
+        self.batch_size = batch_size
+
+def SubMConv3d(in_channels, out_channels, kernel_size, bias=True, indice_key=None, algo=None, **kw):
+    raise NotImplementedError("spconv stub: SubMConv3d not implemented for ZeroGPU")
+
+def SparseConv3d(in_channels, out_channels, kernel_size, stride=1, padding=0, bias=True, indice_key=None, algo=None, **kw):
+    raise NotImplementedError("spconv stub: SparseConv3d not implemented for ZeroGPU")
+
+def SparseInverseConv3d(in_channels, out_channels, kernel_size, indice_key=None, bias=True, algo=None, **kw):
+    raise NotImplementedError("spconv stub: SparseInverseConv3d not implemented for ZeroGPU")
+
+def SparseSequential(*args):
+    return nn.Sequential(*args)