mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-04-23 22:50:17 +08:00
afe9269534
### What problem does this PR solve? Introduced jemalloc. Python uses pymalloc (which is an reimplementation of gblibc malloc) to manage RES. It has pools for small objects to avoid returning memory to OS aggressively. My experience is: Replacing pymalloc with [jemalloc](https://github.com/jemalloc/jemalloc) can reduce RES and speedup task_executor.py. ### Type of change - [x] Performance Improvement
30 lines
706 B
Bash
Executable File
30 lines
706 B
Bash
Executable File
#!/bin/bash
|
|
|
|
# replace env variables in the service_conf.yaml file
|
|
rm -rf /ragflow/conf/service_conf.yaml
|
|
while IFS= read -r line || [[ -n "$line" ]]; do
|
|
# Use eval to interpret the variable with default values
|
|
eval "echo \"$line\"" >> /ragflow/conf/service_conf.yaml
|
|
done < /ragflow/conf/service_conf.yaml.template
|
|
|
|
export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu/
|
|
|
|
PY=python3
|
|
|
|
CONSUMER_NO_BEG=$1
|
|
CONSUMER_NO_END=$2
|
|
|
|
function task_exe(){
|
|
JEMALLOC_PATH=$(pkg-config --variable=libdir jemalloc)/libjemalloc.so
|
|
while [ 1 -eq 1 ]; do
|
|
LD_PRELOAD=$JEMALLOC_PATH $PY rag/svr/task_executor.py $1;
|
|
done
|
|
}
|
|
|
|
for ((i=CONSUMER_NO_BEG; i<CONSUMER_NO_END; i++))
|
|
do
|
|
task_exe $i &
|
|
done
|
|
|
|
wait;
|