9 namespace Autoscheduler {
11 bool all_ones(
const std::vector<int64_t> &nums);
18 const std::vector<int> &vec_dim_serial_sizes,
19 bool filter_small_outer_extents =
false,
20 bool allow_inner_ones =
false);
30 std::vector<std::vector<int64_t>>
generate_tilings(
const std::vector<int64_t> &s,
int d,
int factor,
32 const std::vector<int> &inner_sizes = std::vector<int>());
36 void lowered_dims(
const std::vector<int64_t> &size,
int vector_loop_i, std::vector<int64_t> &lowered_size);
46 std::vector<std::vector<int64_t>>
generate_gpu_tilings(
const std::vector<std::vector<int64_t>> &stage_sizes,
47 const std::vector<std::vector<int>> &pure_dims,
48 const std::vector<int64_t> &max_s,
50 const std::vector<int> &vectorized_indices,
52 bool is_compute_root_stage);