51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
from enum import Enum
|
|
|
|
|
|
class Config:
|
|
class ExperimentCategory(Enum):
|
|
SMALL = [0, 20] # in Mop/experiment
|
|
MEDIUM = [20, 1000]
|
|
LARGE = [1000, 1_000_000]
|
|
VERY_LARGE = [1_000_000, 1_000_000_000]
|
|
|
|
EXPERIMENT_TIME = 1.0
|
|
|
|
ELEMENT_WISE_ARGS = [
|
|
(100, 100),
|
|
(100, 200),
|
|
(128, 128),
|
|
(200, 100),
|
|
(200, 200),
|
|
(256, 256),
|
|
(256, 512),
|
|
(512, 256),
|
|
(400, 400),
|
|
(512, 512),
|
|
(800, 800),
|
|
(1024, 1024),
|
|
(1800, 1800)]
|
|
MATMUL_ARGS = [
|
|
((100, 100), (100, 100)),
|
|
((100, 200), (200, 100)),
|
|
((128, 128), (128, 128)),
|
|
((200, 100), (100, 200)),
|
|
((200, 200), (200, 200)),
|
|
((256, 256), (256, 256)),
|
|
((256, 512), (512, 256)),
|
|
((400, 400), (400, 400)),
|
|
((512, 256), (256, 512)),
|
|
((512, 512), (512, 512)),
|
|
((800, 800), (800, 800)),
|
|
((1000, 1000), (1000, 1000)),
|
|
((1200, 1200), (1200, 1200))]
|
|
NN_1D_ARGS = [
|
|
(1, 16), (16, 16), (64, 16),
|
|
(1, 64), (16, 64),
|
|
(1, 150), (16, 150),
|
|
(1, 256), (16, 256),
|
|
(1, 400), (16, 400), (64, 400),
|
|
(1, 512), (16, 512), (64, 512),
|
|
(1, 800), (16, 800), (64, 800),
|
|
(1, 1024), (16, 1024),
|
|
(1, 2000), (16, 2000), (64, 2000),
|
|
(1, 4000), (16, 4000), (64, 4000)]
|