target.cc

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) NumeriCor GmbH
// SPDX-License-Identifier: LicenseRef-APL-1.1
 
#include <map>
#include <string>
#include <sstream>
 
#include "target.h"
 
namespace limpet {
 
  namespace {
    static std::map<std::string, Target> const map = {
  {"cpu",       Target::CPU},
  {"mlir-cpu",  Target::MLIR_CPU},
  {"mlir-rocm", Target::MLIR_ROCM},
  {"mlir-cuda", Target::MLIR_CUDA},
  {"auto",      Target::AUTO}
};
}
 
Target get_target_from_string(std::string const str) {
  auto t = map.find(str);
  if (t == map.end()) {
    return Target::UNKNOWN;
  }
  else {
    return t->second;
  }
}
 
std::string get_string_from_target(Target const target) {
    for (auto t: map) {
        if (t.second == target) {
            return t.first;
        }
    }
    return "unknown";
}
 
std::string get_target_list_string() {
  std::stringstream list;
  for (auto target : map) {
    list << target.first << ", ";
  }
  // remove last ', '
  return list.str().substr(0, list.str().length() - 2);
}
 
bool is_gpu(Target const target) {
  return target == Target::MLIR_CUDA || target == Target::MLIR_ROCM;
}
 
bool is_concrete(Target const target) {
  return target > Target::UNKNOWN && target < Target::N_TARGETS;
}
 
} // namespace limpet