@@ -21,6 +21,11 @@ const (
2121 interconnectInfiniband = "infiniband"
2222 formFactorSXM5 = "sxm5"
2323 diskTypeSSD = "ssd"
24+
25+ // Currently only 8xH100/H200 instance types are available
26+ // so it's safe to hardcode vCPU and GPU count.
27+ sfcVCPU = 112
28+ sfcGPUCount = 8
2429)
2530
2631func makeDefaultInstanceTypePrice (amount string , currencyCode string ) currency.Amount {
@@ -236,18 +241,14 @@ func getInstanceTypeMetadata(gpuType string) (*sfcInstanceTypeMetadata, error) {
236241 }
237242}
238243
239- // vCPU count provided by SF Compute. Currently only 8xH100/H200 instance types are
240- // available so it's safe to hardcode.
241- const sfcVCPU = 112
242-
243244var h100InstanceTypeMetadata = sfcInstanceTypeMetadata {
244245 gpuType : gpuTypeH100 ,
245246 formFactor : formFactorSXM5 ,
246247 architecture : v1 .ArchitectureX86_64 ,
247248 memoryBytes : v1 .NewBytes (960 , v1 .Gigabyte ),
248249 diskBytes : v1 .NewBytes (1500 , v1 .Gigabyte ),
249250 vcpu : sfcVCPU ,
250- gpuCount : 8 ,
251+ gpuCount : sfcGPUCount ,
251252 gpuManufacturer : v1 .ManufacturerNVIDIA ,
252253 gpuVRAM : v1 .NewBytes (80 , v1 .Gigabyte ),
253254 estimatedDeployTime : 14 * time .Minute ,
@@ -261,7 +262,7 @@ var h200InstanceTypeMetadata = sfcInstanceTypeMetadata{
261262 memoryBytes : v1 .NewBytes (960 , v1 .Gigabyte ),
262263 diskBytes : v1 .NewBytes (1500 , v1 .Gigabyte ),
263264 vcpu : sfcVCPU ,
264- gpuCount : 8 ,
265+ gpuCount : sfcGPUCount ,
265266 gpuManufacturer : v1 .ManufacturerNVIDIA ,
266267 gpuVRAM : v1 .NewBytes (141 , v1 .Gigabyte ),
267268 estimatedDeployTime : 14 * time .Minute ,
0 commit comments