Skip to content
This repository was archived by the owner on Apr 28, 2023. It is now read-only.

Commit 56610d2

Browse files
Jules PondardJules Pondard
authored andcommitted
Add a file to quickly compare different set of options
These options must be given in their vector format. See my_utils for more info
1 parent f9f430a commit 56610d2

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed
Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
import numpy as np
2+
import time
3+
import utils
4+
import torch
5+
import torch.optim as optim
6+
import torch.nn as nn
7+
import torch.utils.data
8+
import torch.nn.functional as F
9+
import tensor_comprehensions as tc
10+
11+
set_options = [
12+
[1, 1, 0, 1, 2, 1, 3, 6, 8, 0, 3, 0, 2, 11, 9, 8, 2, 0, 0, 3, 0, 0, 1, 0, 1, 2]
13+
]
14+
15+
exptuner_config = utils.ExpTunerConfig()
16+
exptuner_config.set_convolution_tc()
17+
18+
for i in range(len(set_options)):
19+
opts = np.array(set_options[i])
20+
time = utils.evalTime(opts, exptuner_config)
21+
print(time)

0 commit comments

Comments
 (0)