halide_utils.h

Go to the documentation of this file.

#pragma once

#include <chrono>
#include <string>
#include <unordered_set>
#include <vector>

#include "tc/core/tc2halide.h"
#include "tc/core/utils/cuda_info.h"
#include "tc/core/utils/dlpack.h"

namespace tc {

std::map<std::string, int> computeParamValueMap(
    const tc2halide::HalideComponents& components,
    const std::vector<const DLTensor*>& inputsDLT);

std::vector<dlutils::DLTensorUPtr> inferOutputTensorInfo(
    const tc2halide::HalideComponents& halide,
    const std::vector<const DLTensor*>& inputsDLT);

std::string halideCodegenC(const Halide::Internal::Stmt& s);

} // namespace tc