blob: 1f62a5c6c9a3be039e8b4829ae27cdcebc3e5b2e (
plain)
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
|
import log_reader
import interactive_host
import sys
from typing import Sequence
def main(args):
# this advisor just picks the first legal register to evict, which is
# identifiable by the "mask" feature
class Advisor:
to_return = False
def advice(self, tensor_values: Sequence[log_reader.TensorValue]):
for tv in tensor_values:
if tv.spec().name != "mask":
continue
for i, v in enumerate(tv):
if v == 1:
return i
# i.e. invalid:
return -1
a = Advisor()
interactive_host.run_interactive(args[0], a.advice, args[1:])
if __name__ == "__main__":
main(sys.argv[1:])
|
