085587e1a9
This reverts commitc3776c11c2. This relands commita959d70eb5. This was originally causing bot failures on Python version 3.8. This relanding fixes that by adjusting the relevant type annotations that are not supported in earlier versions.
29 lines
740 B
Python
29 lines
740 B
Python
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:])
|