Skip to content

Instantly share code, notes, and snippets.

@HahaLan97
Last active April 23, 2024 12:51
Show Gist options
  • Save HahaLan97/d4ef4d4f43c71051a167a7b4d5f48e41 to your computer and use it in GitHub Desktop.
Save HahaLan97/d4ef4d4f43c71051a167a7b4d5f48e41 to your computer and use it in GitHub Desktop.
should be the one
"builtin.module"() ({
"func.func"() ({
^bb0():
%duplicate_60_i_bnd_in = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%duplicate_60_i_bnd_out = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%bcast4_index_i_bnd_0 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%bcast4_index_i_bnd_1 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%bcast4_index_i_bnd_2 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%bcast4_index_i_bnd_3 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%compute_T_prime_T = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%compute_T_prime_j_T = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%compute_T_prime_f_T = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%compute_p_prime_p = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%compute_p_prime_j_strato = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%compute_p_prime_j_p = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%compute_p_prime_f_p = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%compute_k_T = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%compute_k_p = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%compute_k_k = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%bcast4_index_j_T_0 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%bcast4_index_j_T_1 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%bcast4_index_j_T_2 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%bcast4_index_j_T_3 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%bcast3_f32_f_T_0 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%bcast3_f32_f_T_1 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%bcast3_f32_f_T_2 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%bcast4_index_j_strato_0 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%bcast4_index_j_strato_1 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%bcast4_index_j_strato_2 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%bcast4_index_j_strato_3 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<index>)
%bcast3_f32_n_d_in = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%bcast3_f32_n_d_out0 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%bcast3_f32_n_d_out1 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%bcast3_f32_n_d_out2 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%bcast3_gas_in = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<7xf32>>)
%bcast3_gas_out0 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<7xf32>>)
%bcast3_gas_out1 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<7xf32>>)
%bcast3_gas_out2 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<7xf32>>)
%bcast3_gas_out3 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%compute_eta_bnd_r_mix = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<2xf32>>)
%compute_eta_bnd_j_eta = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<2xindex>>)
%compute_eta_bnd_f_eta = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<2xf32>>)
%bcast3_index_2_j_eta_out0 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<2xindex>>)
%bcast3_index_2_j_eta_out1 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<2xindex>>)
%bcast3_index_2_j_eta_out2 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<2xindex>>)
%bcast3_f32_2_f_eta_out0 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<2xf32>>)
%bcast3_f32_2_f_eta_out1 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<2xf32>>)
%bcast3_f32_2_f_eta_out2 = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<2xf32>>)
%tau_major_bnd_tau_g_maj = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<16xf32>>)
%tau_rayleigh_bnd_tau_r = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<16xf32>>)
%dry_fact = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<f32>)
%tau_minor_bnd_tau_g_min = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<16xf32>>)
%tau_sum_tau_g = "olympus.channel"() <{paramType = "small", depth = 840}> : () -> (!olympus.channel<vector<16xf32>>)
"olympus.kernel"(%duplicate_60_i_bnd_in, %duplicate_60_i_bnd_out) <{callee = @duplicate_60, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 1, 1>}> : (!olympus.channel<index>, !olympus.channel<index>) -> ()
"olympus.kernel"(%duplicate_60_i_bnd_out, %bcast4_index_i_bnd_0, %bcast4_index_i_bnd_1, %bcast4_index_i_bnd_2, %bcast4_index_i_bnd_3) <{callee = @bcast4_index, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 1, 4>}> : (!olympus.channel<index>, !olympus.channel<index>, !olympus.channel<index>, !olympus.channel<index>, !olympus.channel<index>) -> ()
"olympus.kernel"(%compute_T_prime_T, %compute_T_prime_j_T, %compute_T_prime_f_T) <{callee = @compute_T_prime, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 1, 2>}> : (!olympus.channel<f32>, !olympus.channel<index>, !olympus.channel<f32>) -> ()
"olympus.kernel"(%compute_p_prime_p, %compute_p_prime_j_strato, %compute_p_prime_j_p, %compute_p_prime_f_p) <{callee = @compute_p_prime, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 1, 3>}> : (!olympus.channel<f32>, !olympus.channel<index>, !olympus.channel<index>, !olympus.channel<f32>) -> ()
"olympus.kernel"(%compute_k_T, %compute_k_p, %compute_k_k) <{callee = @compute_k, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 2, 1>}> : (!olympus.channel<f32>, !olympus.channel<f32>, !olympus.channel<f32>) -> ()
"olympus.kernel"(%compute_T_prime_j_T, %bcast4_index_j_T_0, %bcast4_index_j_T_1, %bcast4_index_j_T_2, %bcast4_index_j_T_3) <{callee = @bcast4_index, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 1, 4>}> : (!olympus.channel<index>, !olympus.channel<index>, !olympus.channel<index>, !olympus.channel<index>, !olympus.channel<index>) -> ()
"olympus.kernel"(%compute_T_prime_f_T, %bcast3_f32_f_T_0, %bcast3_f32_f_T_1, %bcast3_f32_f_T_2) <{callee = @bcast3_f32, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 1, 3>}> : (!olympus.channel<f32>, !olympus.channel<f32>, !olympus.channel<f32>, !olympus.channel<f32>) -> ()
"olympus.kernel"(%compute_p_prime_j_strato, %bcast4_index_j_strato_0, %bcast4_index_j_strato_1, %bcast4_index_j_strato_2, %bcast4_index_j_strato_3) <{callee = @bcast4_index, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 1, 4>}> : (!olympus.channel<index>, !olympus.channel<index>, !olympus.channel<index>, !olympus.channel<index>, !olympus.channel<index>) -> ()
"olympus.kernel"(%bcast3_f32_n_d_in, %bcast3_f32_n_d_out0, %bcast3_f32_n_d_out1, %bcast3_f32_n_d_out2) <{callee = @bcast3_f32, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 1, 3>}> : (!olympus.channel<f32>, !olympus.channel<f32>, !olympus.channel<f32>, !olympus.channel<f32>) -> ()
"olympus.kernel"(%bcast3_gas_in, %bcast3_gas_out0, %bcast3_gas_out1, %bcast3_gas_out2, %bcast3_gas_out3) <{callee = @bcast3_gas, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 1, 4>}> : (!olympus.channel<vector<7xf32>>, !olympus.channel<vector<7xf32>>, !olympus.channel<vector<7xf32>>, !olympus.channel<vector<7xf32>>, !olympus.channel<f32>) -> ()
"olympus.kernel"(%bcast4_index_i_bnd_0, %bcast4_index_j_T_0, %bcast4_index_j_strato_0, %bcast3_f32_n_d_out0, %bcast3_gas_out0, %compute_eta_bnd_r_mix, %compute_eta_bnd_j_eta, %compute_eta_bnd_f_eta) <{callee = @compute_eta_bnd, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 5, 3>}> : (!olympus.channel<index>, !olympus.channel<index>, !olympus.channel<index>, !olympus.channel<f32>, !olympus.channel<vector<7xf32>>, !olympus.channel<vector<2xf32>>, !olympus.channel<vector<2xindex>>, !olympus.channel<vector<2xf32>>) -> ()
"olympus.kernel"(%compute_eta_bnd_j_eta, %bcast3_index_2_j_eta_out0, %bcast3_index_2_j_eta_out1, %bcast3_index_2_j_eta_out2) <{callee = @bcast3_index_2, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 1, 3>}> : (!olympus.channel<vector<2xindex>>, !olympus.channel<vector<2xindex>>, !olympus.channel<vector<2xindex>>, !olympus.channel<vector<2xindex>>) -> ()
"olympus.kernel"(%compute_eta_bnd_f_eta, %bcast3_f32_2_f_eta_out0, %bcast3_f32_2_f_eta_out1, %bcast3_f32_2_f_eta_out2) <{callee = @bcast3_f32_2, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 1, 3>}> : (!olympus.channel<vector<2xf32>>, !olympus.channel<vector<2xf32>>, !olympus.channel<vector<2xf32>>, !olympus.channel<vector<2xf32>>) -> ()
"olympus.kernel"(%bcast4_index_i_bnd_1, %bcast4_index_j_T_1, %bcast3_f32_f_T_0, %bcast4_index_j_strato_1, %compute_p_prime_j_p, %compute_p_prime_f_p, %compute_eta_bnd_r_mix, %bcast3_index_2_j_eta_out0, %bcast3_f32_2_f_eta_out0, %tau_major_bnd_tau_g_maj) <{callee = @tau_major_bnd, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 9, 1>}> : (!olympus.channel<index>, !olympus.channel<index>, !olympus.channel<f32>, !olympus.channel<index>, !olympus.channel<index>, !olympus.channel<f32>, !olympus.channel<vector<2xf32>>, !olympus.channel<vector<2xindex>>, !olympus.channel<vector<2xf32>>, !olympus.channel<vector<16xf32>>) -> ()
"olympus.kernel"(%bcast4_index_i_bnd_2, %bcast4_index_j_T_2, %bcast3_f32_f_T_1, %bcast4_index_j_strato_2, %bcast3_f32_n_d_out1, %bcast3_gas_out1, %bcast3_index_2_j_eta_out1, %bcast3_f32_2_f_eta_out1, %tau_rayleigh_bnd_tau_r) <{callee = @tau_rayleigh_bnd, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 8, 1>}> : (!olympus.channel<index>, !olympus.channel<index>, !olympus.channel<f32>, !olympus.channel<index>, !olympus.channel<f32>, !olympus.channel<vector<7xf32>>, !olympus.channel<vector<2xindex>>, !olympus.channel<vector<2xf32>>, !olympus.channel<vector<16xf32>>) -> ()
"olympus.kernel"(%bcast3_gas_out3, %dry_fact) <{callee = @compute_dry_fact, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 1, 1>}> : (!olympus.channel<f32>, !olympus.channel<f32>) -> ()
"olympus.kernel"(%bcast4_index_i_bnd_3, %bcast4_index_j_T_3, %bcast3_f32_f_T_2, %bcast4_index_j_strato_3, %compute_k_k, %bcast3_f32_n_d_out2, %bcast3_gas_out2, %dry_fact, %bcast3_index_2_j_eta_out2, %bcast3_f32_2_f_eta_out2, %tau_minor_bnd_tau_g_min) <{callee = @tau_minor_bnd, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 14, 1>}> : (!olympus.channel<index>, !olympus.channel<index>, !olympus.channel<f32>, !olympus.channel<index>, !olympus.channel<f32>, !olympus.channel<f32>, !olympus.channel<vector<7xf32>>, !olympus.channel<f32>, !olympus.channel<vector<2xindex>>, !olympus.channel<vector<2xf32>>, !olympus.channel<vector<16xf32>>) -> ()
"olympus.kernel"(%tau_minor_bnd_tau_g_min, %tau_major_bnd_tau_g_maj, %tau_sum_tau_g) <{callee = @tau_sum, evp.path = "./input/kernels.cpp", operandSegmentSizes = array<i32: 2, 1>}> : (!olympus.channel<vector<16xf32>>, !olympus.channel<vector<16xf32>>, !olympus.channel<vector<16xf32>>) -> ()
"func.return"() : () -> ()
}) { function_type = () -> (), sym_name = "top" } : () -> ()
}) : () -> ()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment