Support for H100? (TORCH_CUDA_ARCH_LIST=9.0)

[jkim50104]

I'm trying to run the Minkowski engine in the H100 GPU. However, H100 has TORCH_CUDA_ARCH_LIST=9.0, and it supports from cuda 11.8. I tried running in every pytorch >= 2.0.0, cuda 11.8. I've seen environments working in this pytorch and CUDA version. I assume it is H100's compatibility problem. Did anyone succeed in this setting and GPU?

It compiles well but always asserts this error when I forward in run time.

assertion (in_feat.size(1) == kernel.size(1)) failed. Input feature size and kernel size mismatch

It can be reproduced in the H100 setting with any environment forwarding code.

• OS: Ubuntu 22.04
• Python version: 3.10.5
• Pytorch version: >= 2.0.0
• CUDA version: 11.8
• NVIDIA Driver version: 524
• Minkowski Engine version 0.5.4