Adds accelerator_model as tpuvm if os.environ has TPUVM_MODE (#526)

* Adds check to determine tpu mode by checking env_vars of TPUVM_MODE

* Fixes os env key error

Co-authored-by: Orti Bazar <orti@google.com>

Author: ortigali

perfzero/lib/perfzero/report_utils.py

@@ -18,6 +18,7 @@
 import importlib
 import json
 import logging
+import os
 import perfzero.utils as utils
 import psutil
 import socket
@@ -211,7 +212,9 @@ def build_execution_summary(execution_timestamp, execution_id,
     system_info['platform_name'] = platform_name
   if system_name:
     system_info['system_name'] = system_name
-  if not is_tpu_benchmark:
+  if os.getenv('TPUVM_MODE'):
+    system_info['accelerator_model'] = 'tpuvm'
+  elif not is_tpu_benchmark:
     gpu_info = utils.get_gpu_info()
     if gpu_info:
       system_info['accelerator_driver_version'] = gpu_info['gpu_driver_version']