40 lines
746 B
Go
40 lines
746 B
Go
package pipelines
|
|
|
|
import (
|
|
"github.com/beclab/Olares/cli/pkg/common"
|
|
"github.com/beclab/Olares/cli/pkg/core/connector"
|
|
"github.com/beclab/Olares/cli/pkg/gpu"
|
|
)
|
|
|
|
func GpuDriverStatus() error {
|
|
arg := common.NewArgument()
|
|
|
|
runtime, err := common.NewKubeRuntime(*arg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
runtime.SetRunner(
|
|
&connector.Runner{
|
|
Host: &connector.BaseHost{
|
|
Name: common.LocalHost,
|
|
Arch: runtime.GetSystemInfo().GetOsArch(),
|
|
Os: runtime.GetSystemInfo().GetOsType(),
|
|
},
|
|
},
|
|
)
|
|
|
|
// get gpu status
|
|
if err := new(gpu.PrintGpuStatus).Execute(runtime); err != nil {
|
|
return err
|
|
}
|
|
|
|
// get device plugin status
|
|
if err := new(gpu.PrintPluginsStatus).Execute(runtime); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|