Skip to content

Commit 3f2f21a

Browse files
authored
fix(etl): strip -cw runner-pool suffix in hwToGpuKey (#252)
gb300-cw artifacts from CoreWeave runners were silently dropped because hwToGpuKey did not strip the -cw suffix, leaving "gb300-cw" unresolved.
1 parent 34958c0 commit 3f2f21a

2 files changed

Lines changed: 5 additions & 0 deletions

File tree

packages/db/src/etl/normalizers.test.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,10 @@ describe('hwToGpuKey', () => {
7373
expect(hwToGpuKey('B200-DSV4')).toBe('b200');
7474
});
7575

76+
it('strips -cw suffix', () => {
77+
expect(hwToGpuKey('gb300-cw')).toBe('gb300');
78+
});
79+
7680
it('strips runner index suffix before other suffixes', () => {
7781
expect(hwToGpuKey('mi355x-amd_0')).toBe('mi355x');
7882
expect(hwToGpuKey('mi355x-amd_2')).toBe('mi355x');

packages/db/src/etl/normalizers.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ export function hwToGpuKey(hw: string): string | null {
3535
.replace(/-dgxc$/, '')
3636
.replace(/-nb$/, '')
3737
.replace(/-dsv4$/, '')
38+
.replace(/-cw$/, '')
3839
.replace(/-nv$/, '');
3940
return GPU_KEYS.has(base) ? base : null;
4041
}

0 commit comments

Comments
 (0)