Skip to content
This repository was archived by the owner on Apr 28, 2023. It is now read-only.

Commit 9469abb

Browse files
GeneticAutotunerATen::tune -> startingPoints defaults to {baseMapping}
1 parent 9d48f0a commit 9469abb

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

include/tc/autotuner/genetic_autotuner_aten.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ class GeneticAutotunerATen {
4343
const std::string& tcName,
4444
const std::vector<at::Tensor>& inputs,
4545
CudaMappingOptions baseMapping,
46-
std::vector<CudaMappingOptions> startingPoints,
46+
std::vector<CudaMappingOptions> startingPoints = {},
4747
const TuningParameterFixer& fixedParams = {});
4848

4949
private:

src/autotuner/genetic_autotuner_aten.cc

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,9 @@ llvm::Optional<CudaMappingOptions> GeneticAutotunerATen::tune(
107107
tc::ScopeGuard g1([&]() { deleteGpuDlmTensors(managedInputsPerGpu); });
108108
tc::ScopeGuard g2([&]() { deleteGpuDlmTensors(managedOutputsPerGpu); });
109109

110+
if (startingPoints.size() == 0) {
111+
startingPoints.push_back(baseMapping);
112+
}
110113
return geneticAutotuner_->tune(
111114
cacheFileName,
112115
tcName,

0 commit comments

Comments
 (0)