diff --git a/gen/nvml/generateapi.go b/gen/nvml/generateapi.go index 3d5c8f2..8428b91 100644 --- a/gen/nvml/generateapi.go +++ b/gen/nvml/generateapi.go @@ -190,7 +190,7 @@ func generatePackageMethodsComment(input GeneratableInterfacePoperties) (string, var signature strings.Builder comment := strings.Join(commentFmt, "\n") comment = fmt.Sprintf(comment, input.Type) - signature.WriteString(fmt.Sprintf("%s\n", comment)) + fmt.Fprintf(&signature, "%s\n", comment) return signature.String(), nil } @@ -204,7 +204,7 @@ func generateInterfaceComment(input GeneratableInterfacePoperties) (string, erro var signature strings.Builder comment := strings.Join(commentFmt, "\n") comment = fmt.Sprintf(comment, input.Interface, input.Type, strings.ToLower(input.Interface), input.Interface, input.Interface) - signature.WriteString(fmt.Sprintf("%s\n", comment)) + fmt.Fprintf(&signature, "%s\n", comment) return signature.String(), nil } @@ -232,7 +232,7 @@ func generatePackageMethods(sourceDir string, input GeneratableInterfacePopertie func generateInterface(sourceDir string, input GeneratableInterfacePoperties) (string, error) { var signature strings.Builder - signature.WriteString(fmt.Sprintf("type %s interface {\n", input.Interface)) + fmt.Fprintf(&signature, "type %s interface {\n", input.Interface) methods, err := extractMethodsFromPackage(sourceDir, input) if err != nil { diff --git a/pkg/nvml/mock/dgxa100/mig-profile.go b/pkg/nvml/mock/dgxa100/mig-profile.go index c4df4c8..322c124 100644 --- a/pkg/nvml/mock/dgxa100/mig-profile.go +++ b/pkg/nvml/mock/dgxa100/mig-profile.go @@ -13,7 +13,6 @@ * See the License for the specific language governing permissions and * limitations under the License. */ - package dgxa100 import ( @@ -132,6 +131,17 @@ var MIGProfiles = struct { SharedJpegCount: 0, SharedOfaCount: 0, }, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: { + Id: nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1, + SliceCount: 1, + InstanceCount: 1, + MultiprocessorCount: 14, + SharedCopyEngineCount: 1, + SharedDecoderCount: 0, + SharedEncoderCount: 0, + SharedJpegCount: 0, + SharedOfaCount: 0, + }, }, nvml.GPU_INSTANCE_PROFILE_1_SLICE_REV1: { nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: { @@ -145,6 +155,17 @@ var MIGProfiles = struct { SharedJpegCount: 1, SharedOfaCount: 1, }, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: { + Id: nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1, + SliceCount: 1, + InstanceCount: 1, + MultiprocessorCount: 14, + SharedCopyEngineCount: 1, + SharedDecoderCount: 1, + SharedEncoderCount: 0, + SharedJpegCount: 1, + SharedOfaCount: 1, + }, }, nvml.GPU_INSTANCE_PROFILE_1_SLICE_REV2: { nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: { @@ -158,6 +179,17 @@ var MIGProfiles = struct { SharedJpegCount: 0, SharedOfaCount: 0, }, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: { + Id: nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1, + SliceCount: 1, + InstanceCount: 1, + MultiprocessorCount: 14, + SharedCopyEngineCount: 1, + SharedDecoderCount: 1, + SharedEncoderCount: 0, + SharedJpegCount: 0, + SharedOfaCount: 0, + }, }, nvml.GPU_INSTANCE_PROFILE_2_SLICE: { nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: { @@ -171,6 +203,17 @@ var MIGProfiles = struct { SharedJpegCount: 0, SharedOfaCount: 0, }, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: { + Id: nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1, + SliceCount: 1, + InstanceCount: 2, + MultiprocessorCount: 14, + SharedCopyEngineCount: 2, + SharedDecoderCount: 1, + SharedEncoderCount: 0, + SharedJpegCount: 0, + SharedOfaCount: 0, + }, nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE: { Id: nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE, SliceCount: 2, @@ -195,6 +238,17 @@ var MIGProfiles = struct { SharedJpegCount: 0, SharedOfaCount: 0, }, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: { + Id: nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1, + SliceCount: 1, + InstanceCount: 3, + MultiprocessorCount: 14, + SharedCopyEngineCount: 3, + SharedDecoderCount: 2, + SharedEncoderCount: 0, + SharedJpegCount: 0, + SharedOfaCount: 0, + }, nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE: { Id: nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE, SliceCount: 2, @@ -230,6 +284,17 @@ var MIGProfiles = struct { SharedJpegCount: 0, SharedOfaCount: 0, }, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: { + Id: nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1, + SliceCount: 1, + InstanceCount: 4, + MultiprocessorCount: 14, + SharedCopyEngineCount: 4, + SharedDecoderCount: 2, + SharedEncoderCount: 0, + SharedJpegCount: 0, + SharedOfaCount: 0, + }, nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE: { Id: nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE, SliceCount: 2, @@ -265,6 +330,17 @@ var MIGProfiles = struct { SharedJpegCount: 1, SharedOfaCount: 1, }, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: { + Id: nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1, + SliceCount: 1, + InstanceCount: 7, + MultiprocessorCount: 14, + SharedCopyEngineCount: 7, + SharedDecoderCount: 5, + SharedEncoderCount: 0, + SharedJpegCount: 1, + SharedOfaCount: 1, + }, nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE: { Id: nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE, SliceCount: 2, @@ -438,34 +514,41 @@ var MIGPlacements = struct { // TODO: Fill out ComputeInstancePossiblePlacements ComputeInstancePossiblePlacements: map[int]map[int][]nvml.ComputeInstancePlacement{ nvml.GPU_INSTANCE_PROFILE_1_SLICE: { - nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: {}, }, nvml.GPU_INSTANCE_PROFILE_1_SLICE_REV1: { - nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: {}, }, nvml.GPU_INSTANCE_PROFILE_1_SLICE_REV2: { - nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: {}, }, nvml.GPU_INSTANCE_PROFILE_2_SLICE: { - nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, - nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: {}, + nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE: {}, }, nvml.GPU_INSTANCE_PROFILE_3_SLICE: { - nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, - nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE: {}, - nvml.COMPUTE_INSTANCE_PROFILE_3_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: {}, + nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_3_SLICE: {}, }, nvml.GPU_INSTANCE_PROFILE_4_SLICE: { - nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, - nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE: {}, - nvml.COMPUTE_INSTANCE_PROFILE_4_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: {}, + nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_4_SLICE: {}, }, nvml.GPU_INSTANCE_PROFILE_7_SLICE: { - nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, - nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE: {}, - nvml.COMPUTE_INSTANCE_PROFILE_3_SLICE: {}, - nvml.COMPUTE_INSTANCE_PROFILE_4_SLICE: {}, - nvml.COMPUTE_INSTANCE_PROFILE_7_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_1_SLICE_REV1: {}, + nvml.COMPUTE_INSTANCE_PROFILE_2_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_3_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_4_SLICE: {}, + nvml.COMPUTE_INSTANCE_PROFILE_7_SLICE: {}, }, }, }