host

Author: iisaduan

scripts/src/runTsPerf.ts

@@ -28,6 +28,10 @@ await fn();
 
 async function installHosts() {
     const host = getNonEmptyEnv("TSPERF_JOB_HOST");
+    if (host === "native") {
+        console.log("Skipping host install for native binary");
+        return;
+    }
 
     await $`node ${tsperfExe} host install --host ${host}`;
 }
@@ -59,7 +63,9 @@ async function getCommonBenchmarkArgs() {
         const cpu = getNonEmptyEnv("TSPERF_AGENT_BENCHMARK_CPU");
         const info = await getRepoInfo(args.builtDir);
 
-        tsperfArgs.push("--host", host);
+        if (host !== "native") {
+            tsperfArgs.push("--host", host);
+        }
         tsperfArgs.push("--scenario", scenario);
         tsperfArgs.push("--iterations", iterations);
         tsperfArgs.push("--warmups", warmups);

scripts/src/setupPipeline.ts

@@ -49,6 +49,7 @@ const hosts = {
     node18: "node@18.15.0",
     bun: "bun@1.1.24",
     vscode: "vscode@1.82.1",
+    native: "native", // Native Go binary; no host runtime needed.
 } as const satisfies Record<string, string>;
 
 const allJobKinds = ["tsc", "tsserver", "startup"] as const;
@@ -389,7 +390,7 @@ async function parseInput({ input, isPr, gitParseRev }: SetupPipelineInput) {
     return parsed;
 }
 
-function* transformPreset(parameters: Parameters, iter: Iterable<Scenario>): Iterable<Scenario> {
+function* transformPreset(parameters: Parameters, iter: Iterable<Scenario>, tsgo: boolean): Iterable<Scenario> {
     const all = [...worker()];
 
     for (const scenario of all) {
@@ -410,9 +411,9 @@ function* transformPreset(parameters: Parameters, iter: Iterable<Scenario>): Ite
 
     function* worker(): Iterable<Scenario> {
         for (const scenario of iter) {
-            const hosts = parameters.hosts ?? [scenario.host];
+            const scenarioHosts = tsgo ? [hosts.native] : (parameters.hosts ?? [scenario.host]);
 
-            for (const host of hosts) {
+            for (const host of scenarioHosts) {
                 yield {
                     ...scenario,
                     host,
@@ -433,12 +434,13 @@ export interface SetupPipelineInput {
     input: string;
     baselining: boolean;
     isPr: boolean;
+    tsgo: boolean;
     shouldLog: boolean;
     gitParseRev: (query: string) => Promise<GitParseRevResult>;
 }
 
 export async function setupPipeline(input: SetupPipelineInput) {
-    const { baselining, shouldLog } = input;
+    const { baselining, shouldLog, tsgo } = input;
 
     const parameters = await parseInput(input);
     if (shouldLog) {
@@ -470,7 +472,7 @@ export async function setupPipeline(input: SetupPipelineInput) {
     let maxCost = 0;
     const costPerAgent = new Map<Agent, number>();
 
-    for (const scenario of transformPreset(parameters, preset())) {
+    for (const scenario of transformPreset(parameters, preset(), tsgo)) {
         const agent = baselining ? scenario.agent : "any";
         const jobName = sanitizeJobName(`${scenario.kind}_${scenario.host}_${scenario.name}`);
         matrix[agent][jobName] = {
@@ -591,11 +593,13 @@ if (esMain(import.meta)) {
     const input = getNonEmptyEnv("TSPERF_PRESET");
     const baselining = parseBoolean(process.env.USE_BASELINE_MACHINE, false);
     const isPr = parseBoolean(process.env.IS_PR, false);
+    const tsgo = !!process.env.TSGOFLAG;
 
     const { outputVariables } = await setupPipeline({
         input,
         baselining,
         isPr,
+        tsgo,
         shouldLog: true,
         gitParseRev,
     });