diff --git a/compiler/include/concretelang-c/Support/CompilerEngine.h b/compiler/include/concretelang-c/Support/CompilerEngine.h index 8f97dcd6f..ee7ec9468 100644 --- a/compiler/include/concretelang-c/Support/CompilerEngine.h +++ b/compiler/include/concretelang-c/Support/CompilerEngine.h @@ -53,10 +53,7 @@ MLIR_CAPI_EXPORTED std::string roundTrip(const char *module); MLIR_CAPI_EXPORTED lambdaArgument invokeLambda(lambda l, executionArguments args); -// Initialize and terminate parallelization. Init can be called only once (later -// calls might be ignored by the runtime). You shouldn't reinit after -// termination. -MLIR_CAPI_EXPORTED void initParallelization(); +// Terminate parallelization MLIR_CAPI_EXPORTED void terminateParallelization(); // Create a lambdaArgument from a tensor of different data types diff --git a/compiler/lib/Bindings/Python/CompilerAPIModule.cpp b/compiler/lib/Bindings/Python/CompilerAPIModule.cpp index 39afa9bd5..26dc17199 100644 --- a/compiler/lib/Bindings/Python/CompilerAPIModule.cpp +++ b/compiler/lib/Bindings/Python/CompilerAPIModule.cpp @@ -39,7 +39,6 @@ void mlir::concretelang::python::populateCompilerAPISubmodule( return library(library_path, mlir_modules); }); - m.def("init_parallelization", &initParallelization); m.def("terminate_parallelization", &terminateParallelization); pybind11::class_(m, "JitCompilerEngine") diff --git a/compiler/lib/Bindings/Python/concrete/compiler.py b/compiler/lib/Bindings/Python/concrete/compiler.py index cad8f6830..31fac3584 100644 --- a/compiler/lib/Bindings/Python/concrete/compiler.py +++ b/compiler/lib/Bindings/Python/concrete/compiler.py @@ -9,7 +9,6 @@ from typing import List, Union from mlir._mlir_libs._concretelang._compiler import ( JitCompilerEngine as _JitCompilerEngine, - init_parallelization as _init_parallelization, terminate_parallelization as _terminate_parallelization, ) from mlir._mlir_libs._concretelang._compiler import LambdaArgument as _LambdaArgument @@ -185,9 +184,6 @@ class CompilerEngine: if not isinstance(unsecure_key_set_cache_path, str): raise TypeError("unsecure_key_set_cache_path must be a str") - if any([auto_parallelize, loop_parallelize, df_parallelize]): - # Multiple calls should be guarded in the compiler and only result in a single init - _init_parallelization() self._lambda = self._engine.build_lambda( mlir_str, func_name, diff --git a/compiler/lib/CAPI/Support/CompilerEngine.cpp b/compiler/lib/CAPI/Support/CompilerEngine.cpp index bef38fa68..0aeb76097 100644 --- a/compiler/lib/CAPI/Support/CompilerEngine.cpp +++ b/compiler/lib/CAPI/Support/CompilerEngine.cpp @@ -48,18 +48,9 @@ buildLambda(const char *module, const char *funcName, return std::move(*lambdaOrErr); } -void initParallelization() { -#ifdef CONCRETELANG_PARALLEL_EXECUTION_ENABLED - _dfr_pre_main(); -#else - throw std::runtime_error( - "This package was built without parallelization support"); -#endif -} - void terminateParallelization() { #ifdef CONCRETELANG_PARALLEL_EXECUTION_ENABLED - _dfr_post_main(); + _dfr_terminate(); #else throw std::runtime_error( "This package was built without parallelization support");