1 files changed, 66 insertions, 0 deletions
diff --git a/tools/cuda-ab/p2/src/common.h b/tools/cuda-ab/p2/src/common.h
new file mode 100644
index 0000000..e1e3137
--- /dev/null
+++ b/tools/cuda-ab/p2/src/common.h
@@ -0,0 +1,66 @@
+// SPDX-License-Identifier: BSD-3-Clause
+/*
+ * common.h - shared conventions for the P2 "arbitrary CUDA kernels" suite.
+ *
+ * Line conventions (p2/run_guest.sh filters on these):
+ *
+ *   compared (stable)   RESULT / CHECK / ERR / STATUS / P2PRINT / PROBE / MAIL
+ *   informational       DEVICE / INFO / TIMING / MEMINFO   (ignored by compare)
+ *
+ * Every program accepts --emit-expected: it prints the deterministic
+ * machine-checkable lines a correct GPU run must produce, computed on the
+ * CPU only, without touching CUDA. p2/oracle.sh uses this fallback when the
+ * host has no usable GPU (e.g. the device is owned by a running guest), so
+ * the expected/ references are a documented deterministic computation
+ * rather than a captured bare-metal run.
+ *
+ * All verification in this suite is exact (integers, or bit-identical
+ * float/integer arithmetic and correctly-rounded fmaf), so `bad=0` is
+ * meaningful and no tolerances are used.
+ */
+#ifndef P2_COMMON_H
+#define P2_COMMON_H
+
+#include <cstdio>
+#include <cstdlib>
+#include <cstring>
+#include <cstdint>
+#include <cuda_runtime.h>
+
+#define P2_CHECK(call)                                                        \
+	do {                                                                  \
+		cudaError_t e_ = (call);                                      \
+		if (e_ != cudaSuccess) {                                      \
+			printf("ERR cuda=%s code=%d line=%d\n",               \
+			       cudaGetErrorString(e_), (int)e_, __LINE__);    \
+			printf("STATUS FAIL\n");                              \
+			fflush(stdout);                                       \
+			return 1;                                             \
+		}                                                             \
+	} while (0)
+
+static inline int p2_emit_mode(int argc, char **argv)
+{
+	return (argc > 1 && strcmp(argv[1], "--emit-expected") == 0) ? 1 : 0;
+}
+
+static inline void p2_device_info(void)
+{
+	cudaDeviceProp prop;
+
+	if (cudaGetDeviceProperties(&prop, 0) == cudaSuccess)
+		printf("DEVICE %s cc=%d.%d sm=%d mem=%llu\n", prop.name,
+		       prop.major, prop.minor, prop.multiProcessorCount,
+		       (unsigned long long)prop.totalGlobalMem);
+	else
+		printf("DEVICE unavailable\n");
+}
+
+static inline int p2_status(int bad)
+{
+	printf("STATUS %s\n", bad == 0 ? "OK" : "FAIL");
+	fflush(stdout);
+	return bad == 0 ? 0 : 1;
+}
+
+#endif /* P2_COMMON_H */