dev.py 290 Bytes
Newer Older
yan.yan's avatar
sync  
yan.yan committed
1
2
3
import spconv 


yan.yan's avatar
yan.yan committed
4
5
6
7
from spconv.core_cc.cumm.common import CompileInfo
if __name__ == "__main__":
    print(CompileInfo.arch_is_compatible_gemm((9, 0)), CompileInfo.arch_is_compiled_gemm((9, 0)))
    print(CompileInfo.arch_is_compatible_gemm((8, 6)), CompileInfo.arch_is_compiled_gemm((8, 6)))