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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
"""Benchmark the harmonic_balance function."""
import argparse
import datetime
import socket
import timeit
from timeit import default_timer as timer
import qmix
from qmix.harmonic_balance import harmonic_balance
# Arguments
parser = argparse.ArgumentParser()
parser.add_argument('-s', '--save', action="store_true", help="save speed run?")
args = parser.parse_args()
# Setup ----------------------------------------------------------------------
resp = qmix.respfn.RespFnPolynomial(50, verbose=False)
num_b = (10, 5, 5, 5)
print("\n\tRUNNING SPEED TEST: harmonic_balance\n")
# 1 tone ---------------------------------------------------------------------
cct = qmix.circuit.EmbeddingCircuit(1, 1)
cct.freq[1] = 0.3
cct.vt[1,1] = 0.3
cct.zt[1,1] = 0.5 - 1j*0.3
def one_tone():
harmonic_balance(cct, resp, num_b=num_b, verbose=False)
time_1tone = min(timeit.Timer(one_tone).repeat(50, 1))
print("1 tone:\t\t{:7.3f} ms".format(time_1tone*1e3))
_, it, _ = harmonic_balance(cct, resp, num_b=num_b, verbose=False, mode='x')
print("\t -> {} iterations".format(it))
# 2 tone ---------------------------------------------------------------------
cct = qmix.circuit.EmbeddingCircuit(2, 1)
cct.freq[1] = 0.30
cct.freq[2] = 0.33
cct.vt[1,1] = 0.3
cct.vt[2,1] = 0.03
cct.zt[1,1] = 0.5 - 1j*0.3
cct.zt[2,1] = 0.5 - 1j*0.3
def two_tone():
harmonic_balance(cct, resp, num_b=num_b, verbose=False)
time_2tone = min(timeit.Timer(two_tone).repeat(20, 1))
print("2 tones:\t{:7.3f} ms".format(time_2tone*1e3))
_, it, _ = harmonic_balance(cct, resp, num_b=num_b, verbose=False, mode='x')
print("\t -> {} iterations".format(it))
# 3 tone ---------------------------------------------------------------------
cct = qmix.circuit.EmbeddingCircuit(3, 1)
cct.freq[1] = 0.30
cct.freq[2] = 0.33
cct.freq[3] = 0.27
cct.vt[1,1] = 0.3
cct.vt[2,1] = 0.03
cct.vt[3,1] = 0.03
cct.zt[1,1] = 0.5 - 1j*0.3
cct.zt[2,1] = 0.5 - 1j*0.3
cct.zt[3,1] = 0.5 - 1j*0.3
def three_tone():
harmonic_balance(cct, resp, num_b=num_b, verbose=False)
time_3tone = min(timeit.Timer(two_tone).repeat(3, 1))
print("3 tones:\t{:7.3f} s".format(time_3tone))
_, it, _ = harmonic_balance(cct, resp, num_b=num_b, verbose=False, mode='x')
print("\t -> {} iterations".format(it))
# 4 tone ---------------------------------------------------------------------
cct = qmix.circuit.EmbeddingCircuit(4, 1)
cct.freq[1] = 0.30
cct.freq[2] = 0.33
cct.freq[3] = 0.27
cct.freq[4] = 0.03
cct.vt[1,1] = 0.3
cct.vt[2,1] = 0.03
cct.vt[3,1] = 0.03
cct.zt[1,1] = 0.5 - 1j*0.3
cct.zt[2,1] = 0.5 - 1j*0.3
cct.zt[3,1] = 0.5 - 1j*0.3
cct.zt[4,1] = 1.
# def four_tone():
start = timer()
_, it, _ = harmonic_balance(cct, resp, num_b=num_b, verbose=False, mode='x')
time_4tone = timer() - start
print("4 tones:\t{:7.3f} s".format(time_4tone))
print("\t -> {} iterations".format(it))
print("")
# WRITE TO FILE --------------------------------------------------------------
if args.save:
with open('results/harmonic-balance.txt', 'a') as f:
now = datetime.datetime.now()
machine = socket.gethostname()
f.write("{}\t{:.6f}\t{:.6f}\t{:.6f}\t{:.6f}\t{}\n".format(now,
time_1tone,
time_2tone,
time_3tone,
time_4tone,
machine))
|
