Skip to content

File tree

3 files changed

+4
-1
lines changed

3 files changed

+4
-1
lines changed

torchci/lib/benchmark/api_helper/backend/dataFetchers/fetchers.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,13 +37,15 @@ const metaCtors: Record<string, new () => BenchmarkMetadataFetcher> = {
3737
pytorch_operator_microbenchmark: PytorchOperatorMicrobenchmarkMetadataFetcher,
3838
torchao_micro_api_benchmark: TorchAoMicrobApienchmarkMetadataFetcher,
3939
vllm_benchmark: VllmBenchmarkMetadataFetcher,
40+
pytorch_x_vllm_benchmark: VllmBenchmarkMetadataFetcher,
4041
default: BenchmarkMetadataQuery,
4142
};
4243

4344
// Register benchmark list commit fetchers. this is mainly used in list_commits api
4445
const listCommitsCtors: Record<string, new () => BenchmarkListCommitFetcher> = {
4546
pytorch_operator_microbenchmark: PytorchOperatorMicroListCommitsDataFetcher,
4647
vllm_benchmark: VllmListCommitsDataFetcher,
48+
pytorch_x_vllm_benchmark: VllmListCommitsDataFetcher,
4749
default: BenchmarkListCommitQueryBuilder,
4850
};
4951

torchci/lib/benchmark/api_helper/backend/dataFetchers/queryBuilderUtils/benchmarkDataQueryBuilder.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -753,6 +753,7 @@ export class VllmBenchmarkDataFetcher
753753
];
754754
const params = {
755755
...inputs,
756+
modelCategory: inputs.modelCategory ?? "",
756757
excludedMetrics: excludedMetrics,
757758
};
758759

torchci/lib/benchmark/api_helper/backend/dataFetchers/queryBuilderUtils/listCommitQueryBuilder.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ export class VllmListCommitsDataFetcher
174174
toQueryParams(inputs: any, id?: string): Record<string, any> {
175175
const pq = {
176176
...inputs,
177-
operatorName: inputs.operatorName ?? "",
177+
modelCategory: inputs.modelCategory ?? "",
178178
};
179179
const params = this._data_query.toQueryParams(pq);
180180
return params;

0 commit comments

Comments
 (0)