Skip to content

Commit

Permalink
feat: [SKU modularization] sku handler interface (#357)
Browse files Browse the repository at this point in the history
**Reason for Change**:
Adding a CloudSKUHandler interface that will be used to modularize gpu
skus across multiple cloud platforms

**Requirements**

- [ ] added unit tests and e2e tests (if applicable).

**Issue Fixed**:
<!-- If this PR fixes GitHub issue 4321, add "Fixes #4321" to the next
line. -->

**Notes for Reviewers**:
The cloud specific implementations and fixes/updates to current code
will be part of future PRs.
  • Loading branch information
smritidahal653 authored Apr 25, 2024
1 parent 0b63598 commit aa89d20
Showing 1 changed file with 16 additions and 0 deletions.
16 changes: 16 additions & 0 deletions pkg/sku/cloud_sku_handler.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT license.

package sku

type CloudSKUHandler interface {
GetSupportedSKUs() []string
GetGPUConfigs() map[string]GPUConfig
}

type GPUConfig struct {
SKU string
GPUCount int
GPUMem int
GPUModel string
}

0 comments on commit aa89d20

Please sign in to comment.