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

Reduce unrolling factors during tuning #354

Merged
merged 2 commits into from
Apr 27, 2018
Merged

Reduce unrolling factors during tuning #354

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
db43d47
Reduce unrolling factors during tuning
nicolasvasilache Apr 26, 2018
59ca04f
Delete unused rangeUpTo64
nicolasvasilache Apr 26, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 1 addition & 14 deletions tc/autotuner/genetic_tuning_harness.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -110,17 +110,6 @@ void GeneticTunerHarness::stopAfterCurrentGeneration() {

namespace {

std::vector<size_t> filterHigherThan(
const std::vector<size_t>& v,
size_t limit) {
std::vector<size_t> newV;
std::copy_if(
v.begin(), v.end(), std::back_inserter(newV), [limit](size_t val) {
return val <= limit;
});
return newV;
}

void removeDuplicates(std::vector<size_t>& v) {
std::sort(v.begin(), v.end());
v.erase(std::unique(v.begin(), v.end()), v.end());
Expand Down Expand Up @@ -159,7 +148,6 @@ size_t largestDim(const std::vector<const DLTensor*>& inputs) {
void GeneticTunerHarness::setupTuningParameters() {
CHECK_GT(kInputs_.size(), 0u);
auto range = inputDivisorsAndPowers2(kInputs_.begin()->second);
auto rangeUpTo64 = filterHigherThan(range, 64);

// 0 is a valid tiling annotation and signals no tiling of that dimension
// 0 is not a valid block / grid annotation
Expand All @@ -172,8 +160,7 @@ void GeneticTunerHarness::setupTuningParameters() {
configuration.blockParams.setRange(range, "b");
configuration.gridParams.setRange(range, "g");

configuration.unrollFactor =
RangeParameter({1, 2, 4, 8, 16, 32, 64, 128, 256}, "unroll");
configuration.unrollFactor = RangeParameter({1, 2, 4, 8, 16, 32}, "unroll");
}

CudaMappingOptions GeneticTunerHarness::makeOptions(
Expand Down