github/intel-device-plugins-for-kubernetes
7
0
Fork 0
mirror of https://github.com/intel/intel-device-plugins-for-kubernetes.git synced 2025-06-03 03:59:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

gpu: labeler: convert private getter functions into public

Browse Source 
Signed-off-by: Tuomas Katila <tuomas.katila@intel.com>
This commit is contained in:
Tuomas Katila 2023-11-09 17:14:08 +02:00
parent 5b7ec1faa2
commit ef16dc0e9d
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
cmd/internal/labeler/labeler.go
View File

@ -163,10 +163,10 @@ func fallback() uint64 {
return getEnvVarNumber(memoryOverrideEnv) return getEnvVarNumber(memoryOverrideEnv)
} }
func (l *labeler) getMemoryAmount(gpuName string, numTiles uint64) uint64 { func GetMemoryAmount(sysfsDrmDir, gpuName string, numTiles uint64) uint64 {
reserved := getEnvVarNumber(memoryReservedEnv) reserved := getEnvVarNumber(memoryReservedEnv)
filePath := filepath.Join(l.sysfsDRMDir, gpuName, "lmem_total_bytes") filePath := filepath.Join(sysfsDrmDir, gpuName, "lmem_total_bytes")
dat, err := os.ReadFile(filePath) dat, err := os.ReadFile(filePath)
if err != nil { if err != nil {
@ -183,9 +183,9 @@ func (l *labeler) getMemoryAmount(gpuName string, numTiles uint64) uint64 {
return totalPerTile*numTiles - reserved return totalPerTile*numTiles - reserved
} }
// getTileCount reads the tile count. // GetTileCount reads the tile count.
func (l *labeler) getTileCount(gpuName string) (numTiles uint64) { func GetTileCount(sysfsDrmDir, gpuName string) (numTiles uint64) {
filePath := filepath.Join(l.sysfsDRMDir, gpuName, "gt/gt*") filePath := filepath.Join(sysfsDrmDir, gpuName, "gt/gt*")
files, _ := filepath.Glob(filePath) files, _ := filepath.Glob(filePath)
@ -196,9 +196,9 @@ func (l *labeler) getTileCount(gpuName string) (numTiles uint64) {
return uint64(len(files)) return uint64(len(files))
} }
// getNumaNode reads the cards numa node. // GetNumaNode reads the cards numa node.
func (l *labeler) getNumaNode(gpuName string) int { func GetNumaNode(sysfsDrmDir, gpuName string) int {
filePath := filepath.Join(l.sysfsDRMDir, gpuName, "device/numa_node") filePath := filepath.Join(sysfsDrmDir, gpuName, "device/numa_node")
data, err := os.ReadFile(filePath) data, err := os.ReadFile(filePath)
if err != nil { if err != nil {
@ -295,14 +295,14 @@ func (l *labeler) createLabels() error {
return errors.Wrap(err, "gpu name parsing error") return errors.Wrap(err, "gpu name parsing error")
} }
numTiles := l.getTileCount(gpuName) numTiles := GetTileCount(l.sysfsDRMDir, gpuName)
tileCount += int(numTiles) tileCount += int(numTiles)
memoryAmount := l.getMemoryAmount(gpuName, numTiles) memoryAmount := GetMemoryAmount(l.sysfsDRMDir, gpuName, numTiles)
gpuNumList = append(gpuNumList, gpuName[4:]) gpuNumList = append(gpuNumList, gpuName[4:])
// get numa node of the GPU // get numa node of the GPU
numaNode := l.getNumaNode(gpuName) numaNode := GetNumaNode(l.sysfsDRMDir, gpuName)
if numaNode >= 0 { if numaNode >= 0 {
// and store the gpu under that node id // and store the gpu under that node id