Skip to content

Commit a623686

Browse files
authored
[CINN]Call GetKernelInfo in thread to speed up compile (#64893) (#65056)
* Call GetKernelInfo in thread to speed up compile * Add comment
1 parent cee8578 commit a623686

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddle/cinn/hlir/framework/pir_compiler.cc

+2
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,8 @@ std::vector<pir::CINNKernelInfo> PirCompiler::Build(
7777
auto worker_fn = [&](int index) {
7878
CompilationTask task(&group_compilation_contexts[index]);
7979
compilation_results[index] = task();
80+
// Triggering llvm compilation in thread
81+
compilation_results[index]->GetKernelInfo();
8082
};
8183
utils::parallel_run(worker_fn,
8284
utils::SequenceDispatcher(0, task_size),

0 commit comments

Comments
 (0)