Class CUtensorMap

examples/matmul/this-sm90/test_v6.py:9–10

import subprocess, ctypes, torch, os

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

BM, BN, BK = 128, 256, 64
CLUSTER_SIZE = 2