Class CUtensorMap

examples/matmul/this-sm90/test_v5.py:5–6

import os, subprocess, ctypes, torch, gc

class CUtensorMap(ctypes.Structure):
    _fields_ = [("data", ctypes.c_ubyte * 128)]

BM, BN, BK, CS = 128, 128, 64, 2