| 133 | } |
| 134 | |
| 135 | void MLIRCUDAExecutable::execute(JITExecutor* fusion_opr) { |
| 136 | FuncCache* func; |
| 137 | auto cn = fusion_opr->comp_node(); |
| 138 | auto&& prop = CompNodeEnv::from_comp_node(cn).cuda_env().device_prop; |
| 139 | func = &m_func_cache[{prop.major, prop.minor}]; |
| 140 | func->kernel_data = m_kernel_data; |
| 141 | func->exec(fusion_opr, this); |
| 142 | } |
| 143 | |
| 144 | MLIRCUDAExecutable::~MLIRCUDAExecutable() {} |
| 145 |