"fs/git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "5e20b170a783c8a476991ca4a7ee32029761fe66"
Unverified Commit c71dadba authored by SparkSnail's avatar SparkSnail Committed by GitHub
Browse files

Add log information for gpu collector (#3506)

parent d6b5c6ca
...@@ -79,6 +79,7 @@ export function parseGpuIndices(gpuIndices?: string): Set<number> | undefined { ...@@ -79,6 +79,7 @@ export function parseGpuIndices(gpuIndices?: string): Set<number> | undefined {
export const GPU_INFO_COLLECTOR_FORMAT_WINDOWS: string = export const GPU_INFO_COLLECTOR_FORMAT_WINDOWS: string =
` `
$env:METRIC_OUTPUT_DIR="{0}" $env:METRIC_OUTPUT_DIR="{0}"
$app = Start-Process "python" -ArgumentList "-m nni.tools.gpu_tool.gpu_metrics_collector" -passthru -NoNewWindow $app = Start-Process "python" -ArgumentList "-m nni.tools.gpu_tool.gpu_metrics_collector" -passthru -NoNewWindow \
-redirectStandardOutput {0}\\stdout -redirectStandardError {0}\\stderr
Write $app.ID | Out-File {1} -NoNewline -encoding utf8 Write $app.ID | Out-File {1} -NoNewline -encoding utf8
`; `;
...@@ -236,7 +236,8 @@ export function getScriptName(fileNamePrefix: string): string { ...@@ -236,7 +236,8 @@ export function getScriptName(fileNamePrefix: string): string {
} }
export function getGpuMetricsCollectorBashScriptContent(scriptFolder: string): string { export function getGpuMetricsCollectorBashScriptContent(scriptFolder: string): string {
return `echo $$ > ${scriptFolder}/pid ; METRIC_OUTPUT_DIR=${scriptFolder} python3 -m nni.tools.gpu_tool.gpu_metrics_collector`; return `echo $$ > ${scriptFolder}/pid ; METRIC_OUTPUT_DIR=${scriptFolder} python3 -m nni.tools.gpu_tool.gpu_metrics_collector \
1>${scriptFolder}/stdout 2>${scriptFolder}/stderr`;
} }
export function runGpuMetricsCollector(scriptFolder: string): void { export function runGpuMetricsCollector(scriptFolder: string): void {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment