"...composable_kernel_rocm.git" did not exist on "9ec586fc6884fcff4b91a909fb0f866001a5cb39"
Unverified Commit 85c0d841 authored by Lijiaoa's avatar Lijiaoa Committed by GitHub
Browse files

Support show customized tuner/assessor name and get optimize_mode from...

Support show customized tuner/assessor name and get optimize_mode from tuner/advisor/assessor (#3317)
parent fe974e6b
......@@ -11,26 +11,17 @@ export const Command1 = (): any => {
const builtinName: string[] = [];
if (tuner !== undefined) {
title.push('Tuner');
if (tuner.builtinTunerName !== undefined) {
builtinName.push(tuner.builtinTunerName);
}
builtinName.push(tuner.builtinTunerName || tuner.className || 'unknown');
}
if (advisor !== undefined) {
title.push('Advisor');
if (advisor.builtinAdvisorName !== undefined) {
builtinName.push(advisor.builtinAdvisorName);
}
if (advisor.className !== undefined) {
builtinName.push(advisor.className);
}
builtinName.push(advisor.builtinAdvisorName || advisor.className || 'unknown');
}
if (assessor !== undefined) {
title.push('Assessor');
if (assessor.builtinAssessorName !== undefined) {
builtinName.push(assessor.builtinAssessorName);
}
builtinName.push(assessor.builtinAssessorName || assessor.className || 'unknown');
}
return (
......
......@@ -128,7 +128,19 @@ class Experiment {
get optimizeMode(): string {
const tuner = this.profile.params.tuner;
return tuner && tuner.classArgs && tuner.classArgs.optimize_mode ? tuner.classArgs.optimize_mode : 'unknown';
const advisor = this.profile.params.advisor;
const assessor = this.profile.params.assessor;
const resultTuner =
tuner && tuner.classArgs && tuner.classArgs.optimize_mode ? tuner.classArgs.optimize_mode : undefined;
const resultAdvisor =
advisor && advisor.classArgs && advisor.classArgs.optimize_mode
? advisor.classArgs.optimize_mode
: undefined;
const resultAssessor =
assessor && assessor.classArgs && assessor.classArgs.optimize_mode
? assessor.classArgs.optimize_mode
: undefined;
return resultTuner || resultAdvisor || resultAssessor || 'unknown';
}
get trainingServicePlatform(): string {
......
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