1 | #include "taichi/runtime/llvm/launch_arg_info.h" |
---|---|
2 | |
3 | #include "taichi/program/kernel.h" |
4 | |
5 | namespace taichi::lang { |
6 | |
7 | bool LlvmLaunchArgInfo::operator==(const LlvmLaunchArgInfo &other) const { |
8 | return is_array == other.is_array; |
9 | } |
10 | |
11 | std::vector<LlvmLaunchArgInfo> infer_launch_args(const Kernel *kernel) { |
12 | std::vector<LlvmLaunchArgInfo> res; |
13 | res.reserve(kernel->parameter_list.size()); |
14 | for (const auto &a : kernel->parameter_list) { |
15 | res.push_back(LlvmLaunchArgInfo{a.is_array}); |
16 | } |
17 | return res; |
18 | } |
19 | |
20 | } // namespace taichi::lang |
21 |