import enum import ck_types from copy import deepcopy from dataclasses import dataclass from enum import auto from typing import List import os.path import shutil import functools import operator import collections import subprocess import re import gemm_op from gemm_op import * import user from ck_types import * from gemm_ex import * # holds multiple gemm instances op_collection = user.CreateGemmOperator() # emit for each instance for op in op_collection: x = EmitGemmInstance() x.emit(op)