driver.py 533 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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()
Astha Rai's avatar
Astha Rai committed
22
23

# emit for each instance
24
25
26
27
for op in op_collection:
    x = EmitGemmInstance()
    x.emit(op)