memory_promotion_heuristic.h

Go to the documentation of this file.

#pragma once

#include <cstddef>
#include <vector>

#include "tc/external/isl.h"

namespace tc {
namespace polyhedral {
using ThreadIdxxScheduleDepthState =
    std::vector<std::pair<isl::union_set, size_t>>;

class MappedScop;

// In the given mapped scop "mscop",
// promote to shared memory at "depth" until "sharedMemorySize" is used.
// Map copies between global and shared memory to threads and unroll those
// copies if "unrollCopies" is set, using the options in "mscop".
// "threadIdxxScheduleDepthState" contains the schedule depth at which the
// computation was mapped to thread x and is used to check whether the global
// memory is accessed in a coalesced way.
void promoteGreedilyAtDepth(
    MappedScop& scop,
    const ThreadIdxxScheduleDepthState& threadIdxxScheduleDepthState,
    std::size_t depth,
    std::size_t sharedMemorySize,
    bool unrollCopies);
} // namespace polyhedral
} // namespace tc