Update tools/run_benchmark.py to new gradle setup

This makes it possible to run the benchmarks locally. Not sure
if this is enough to make them work on Golem.

Test: tools/run_benchmark.py --target d8 --benchmark HelloWorldDex .
Test: tools/run_benchmark.py --nolib --target d8 --benchmark HelloWorldDex .

Bug: b/302999071
Change-Id: Ibc28aaaebfcae60ae0b07f7ed6a3dafa9456cc3e
diff --git a/tools/utils.py b/tools/utils.py
index 4b15548..87c9fe0 100644
--- a/tools/utils.py
+++ b/tools/utils.py
@@ -53,6 +53,7 @@
 GRADLE_TASK_SOURCE_JAR = ':test:sourcesJar'
 GRADLE_TASK_SWISS_ARMY_KNIFE = ':main:swissArmyKnife'
 GRADLE_TASK_TEST = ':test:test'
+GRADLE_TASK_ALL_TESTS_WITH_APPLY_MAPPING_JAR = ':test:allTestsWithApplyMapping'
 GRADLE_TASK_TEST_DEPS_JAR = ':test:allDepsJar'
 GRADLE_TASK_TEST_JAR = ':test:allTestsJar'