31 static std::string get_env(
const char *v) {
38 static int32_t get_env_i32(
const char *v) {
39 return std::atoi(get_env(v).
c_str());
42 const size_t total_shards, shard_index;
60 : total_shards(get_env_i32(
"TEST_TOTAL_SHARDS")),
61 shard_index(get_env_i32(
"TEST_SHARD_INDEX")) {
64 if (total_shards != 0) {
66 std::cerr <<
"Illegal values for sharding: total " << total_shards <<
" current " << shard_index <<
"\n";
73 if (total_shards > 0) {
74 return (
task_index % total_shards) == shard_index;
81 return total_shards > 0;
static void accept_sharded_status()
bool should_run(size_t task_index) const
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
@ Internal
Not visible externally, similar to 'static' linkage in C.
Internal::ConstantInterval cast(Type t, const Internal::ConstantInterval &a)
Cast operators for ConstantIntervals.
signed __INT32_TYPE__ int32_t
char * getenv(const char *)