audit: bind scale evidence to authorized GPUs

1 files changed, 5 insertions, 1 deletions

diff --git a/experiments/analyze_oral_a_dynamic_scaling.py b/experiments/analyze_oral_a_dynamic_scaling.py
index 38ce967..972356a 100755
--- a/experiments/analyze_oral_a_dynamic_scaling.py
+++ b/experiments/analyze_oral_a_dynamic_scaling.py
@@ -102,6 +102,10 @@ def validate_record(record, path, method, depth, seed):
             and evaluation.get("test_evaluations") == 1
             and evaluation.get("test_used_for_selection") is False,
             f"{path}: evaluation drift")
+    hardware = record.get("hardware", {})
+    require(hardware.get("cuda_device_name") == "NVIDIA GeForce GTX 1080"
+            and hardware.get("cuda_visible_devices") in ("5", "7"),
+            f"{path}: unauthorized hardware")
     epochs = record.get("epochs", [])
     require(len(epochs) == 200
             and all(row.get("epoch") == index + 1
@@ -129,7 +133,7 @@ def validate_record(record, path, method, depth, seed):
         "elementwise_operations": int(record["work"][
             "elementwise_operations_estimate"]),
         "logical_queries": int(record["work"]["logical_batch_loss_queries"]),
-        "peak_memory": int(record["hardware"][
+        "peak_memory": int(hardware[
             "peak_memory_allocated_bytes"]),
         "source_commit": record["provenance"]["git_commit"],
     }