1 | #include "taichi/ir/ir_builder.h" |
---|---|
2 | #include "taichi/ir/statements.h" |
3 | #include "taichi/program/program.h" |
4 | |
5 | void run_snode(); |
6 | void autograd(); |
7 | void aot_save(taichi::Arch arch); |
8 | |
9 | int main() { |
10 | run_snode(); |
11 | autograd(); |
12 | #ifdef TI_WITH_VULKAN |
13 | aot_save(taichi::Arch::vulkan); |
14 | #endif |
15 | #ifdef TI_WITH_DX12 |
16 | aot_save(taichi::Arch::dx12); |
17 | #endif |
18 | return 0; |
19 | } |
20 |