gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

More efficient parameters

Browse Source
This commit is contained in:
Philippe Tillet
2014-09-15 18:30:30 -04:00
parent 64344f3a0a
commit 878cefa29b
2 changed files with 89 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
autotune/python/optimize.py
View File

@@ -7,10 +7,7 @@ import itertools
import tools
import deap.tools
from deap import base
from deap import creator
from genetic import GeneticOperators
from genetic import eaMuPlusLambda
def exhaustive(statement, context, TemplateType, build_template, parameter_names, all_parameters, compute_perf, perf_metric, out):
device = context.devices[0]
@@ -39,25 +36,4 @@ def exhaustive(statement, context, TemplateType, build_template, parameter_names
def genetic(statement, context, TemplateType, build_template, parameter_names, all_parameters, compute_perf, perf_metric, out):
GA = GeneticOperators(context.devices[0], statement, all_parameters, parameter_names, TemplateType, build_template)
creator.create("FitnessMin", base.Fitness, weights=(-1.0,))
creator.create("Individual", list, fitness=creator.FitnessMin)
toolbox = base.Toolbox()
toolbox.register("individual", deap.tools.initIterate, creator.Individual, GA.init)
toolbox.register("population", deap.tools.initRepeat, list, toolbox.individual)
toolbox.register("evaluate", GA.evaluate)
toolbox.register("mate", deap.tools.cxTwoPoint)
toolbox.register("mutate", GA.mutate)
toolbox.register("select", deap.tools.selBest)
pop = toolbox.population(n=50)
hof = deap.tools.HallOfFame(1)
best_performer = lambda x: max([compute_perf(hof[0].fitness.values[0]) for t in x])
best_profile = lambda x: '(%s)'%','.join(map(str,hof[0]))
stats = deap.tools.Statistics(lambda ind: ind.fitness.values)
stats.register("max (" + perf_metric + ")", lambda x: max([compute_perf(hof[0].fitness.values[0]) for t in x]))
stats.register("profile ", lambda x: '(%s)'%','.join(map(str,hof[0])))
pop = eaMuPlusLambda(pop, toolbox, 50, 70, cxpb=0.2, mutpb=0.3, maxtime='5m0s', maxgen=1000, halloffame=hof, compute_perf=compute_perf, perf_metric=perf_metric)
GA.optimize(maxtime='5m0s', maxgen=1000, compute_perf=compute_perf, perf_metric=perf_metric)