flags.h

Go to the documentation of this file.

#pragma once

#include <gflags/gflags.h>
#include <glog/logging.h>

namespace tc {

//
// Declare global general flags
//
DECLARE_bool(debug_lang);
DECLARE_bool(debug_halide);
DECLARE_bool(debug_tc_mapper);
DECLARE_bool(debug_cuda);
DECLARE_bool(debug_tuner);
DECLARE_bool(dump_cuda);

// llvm codegen
DECLARE_bool(llvm_dump_ir);

// Used in benchmarking and autotuning
DECLARE_uint32(benchmark_warmup);
DECLARE_uint32(benchmark_iterations);

// Used in autotuning
DECLARE_uint32(tuner_gen_pop_size);
DECLARE_uint32(tuner_gen_crossover_rate);
DECLARE_uint32(tuner_gen_mutation_rate);
DECLARE_uint32(tuner_gen_generations);
DECLARE_uint32(tuner_gen_number_elites);
DECLARE_uint32(tuner_threads);
DECLARE_string(tuner_gpus);
DECLARE_bool(tuner_print_best);
DECLARE_string(tuner_proto);
DECLARE_string(tuner_rng_restore);
DECLARE_bool(tuner_gen_restore_from_proto);
DECLARE_uint32(tuner_gen_restore_number);
DECLARE_bool(tuner_gen_log_generations);
DECLARE_uint64(tuner_min_launch_total_threads);

// Misc
DECLARE_int64(random_seed);
DECLARE_bool(schedule_tree_verbose_validation);

// random seed setting for reproducibility and debugging purposes
uint64_t initRandomSeed();
const uint64_t& randomSeed();

// python
namespace python {
bool globalDebugGflagsGlogInit(int* pargc, char*** pargv);
} // namespace python

} // namespace tc