driver.py 535 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
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 *
#from make_template import *

# holds multiple gemm instances
op_collection = user.CreateGemmOperator()
for op in op_collection:
    x = EmitGemmInstance()
    x.emit(op)