diff --git a/TEST_MAPPING b/TEST_MAPPING
index 9a0667a..9d84571 100644
--- a/TEST_MAPPING
+++ b/TEST_MAPPING
@@ -9,6 +9,16 @@
       "name": "libsurfaceflinger_unittest"
     }
   ],
+  "mainline-presubmit": [
+    {
+      "name": "CtsPerfettoTestCases[com.google.android.art.apex]",
+      "options": [
+        {
+          "include-filter": "HeapprofdJavaCtsTest*"
+        }
+      ]
+    }
+  ],
   "hwasan-presubmit": [
     {
       "name": "CtsPerfettoTestCases"
diff --git a/test/cts/AndroidTest.xml b/test/cts/AndroidTest.xml
index 1c5c25b..84e72bc 100644
--- a/test/cts/AndroidTest.xml
+++ b/test/cts/AndroidTest.xml
@@ -20,6 +20,7 @@
     <option name="config-descriptor:metadata" key="parameter" value="not_multi_abi" />
     <option name="config-descriptor:metadata" key="parameter" value="secondary_user" />
     <option name="test-suite-tag" value="cts" />
+    <option name="config-descriptor:metadata" key="mainline-param" value="com.google.android.art.apex" />
     <target_preparer class="com.android.tradefed.targetprep.suite.SuiteApkInstaller">
         <option name="cleanup-apks" value="true" />
         <option name="test-file-name" value="CtsPerfettoProducerApp.apk" />