| import log_reader | |
| import interactive_host | |
| import sys | |
| 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: list[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:]) |