def monitoring_helper(service_addr, duration_ms, monitoring_level, num_queries): """Helper function to print monitoring results. Helper function to print monitoring results for num_queries times. Args: service_addr: Address of the TPU profiler service. duration_ms: Duration of one monitoring sample in milliseconds. monitoring_level: An integer between 1 and 2. Level 2 is more verbose than level 1 and shows more metrics. num_queries: Number of monitoring samples to collect. """ if monitoring_level <= 0 or monitoring_level > 2: sys.exit('Please choose a monitoring level between 1 and 2.') for query in range(0, num_queries): res = profiler_client.monitor(service_addr, duration_ms, monitoring_level) print('Cloud TPU Monitoring Results (Sample ', query, '):\n\n', res)
def testMonitor_ProcessInvalidAddress(self): with self.assertRaises(errors.UnavailableError): profiler_client.monitor('localhost:6006', 2000)
def tpu_utilization(cls, service_addr, duration_ms, monitoring_level): return profiler_client.monitor(service_addr, duration_ms, monitoring_level)
def testMonitor_ProcessInvalidAddress(self): # Monitor is only supported in cloud TPU. Test invalid address instead. with self.assertRaises(errors.UnavailableError): profiler_client.monitor('localhost:6006', 2000)