[INSTRUMENTATION] Generalize code in test_gpuhello.py
#5263
+3
−4
test_gpuhello.py
#5263