adding villar light curve fitting using batch GPU

[frenbox]

Trying to fit lightcurves to Villar model as a batch using GPU for ZTF sources

[frenbox]

Build and test the GPU version

docker build --build-arg USE_GPU=true --target dev -t boom-dev --no-cache .
docker compose --profile api up -d

docker run --rm -it --gpus all --network boom_boom   -v $(pwd)/config.yaml:/app/config.yaml   -v $(pwd)/data:/app/data   -v $(pwd)/.env:/app/.env   boom-dev bash

Inside bash
(the features gpu is not really necessary for producer and consumer)

cargo run --release --features gpu --bin kafka_producer ztf 20240617 public --server-url broker:29092

cargo run --release --features gpu --bin kafka_consumer ztf 20240617 --programids public

cargo run --release --features gpu --bin scheduler ztf --config config.yaml

Example of an output

[Theodlz]

Build and test the GPU version

docker build --build-arg USE_GPU=true --target dev -t boom-dev --no-cache .
docker compose --profile api up -d

docker run --rm -it --gpus all --network boom_boom   -v $(pwd)/config.docker.yaml:/app/config.yaml   -v $(pwd)/data:/app/data   -v $(pwd)/.env:/app/.env   boom-dev-gpu bash

Inside bash (the features gpu is not really necessary for producer and consumer)

cargo run --release --features gpu --bin kafka_producer ztf 20240617 public --server-url broker:29092

cargo run --release --features gpu --bin kafka_consumer ztf 20240617 --programids public

cargo run --release --features gpu --bin scheduler ztf --config config.yaml

Example of an output

How does one use the villar_fit array to select candidates?

[frenbox]

I have to name them properly. That part is missing :). I spent the whole day figuring out various CUDA version and how it interacts with ONNX. I will get that done tomorrow morning.

In short, villar fit params represent the 7 parameter for g and r band for this equation

Reference: paper

You have signature rise time, fall time, peak magnitude, etc (in a normalized space .... will need to change in back properly. So final touchup remains)

[frenbox]

Build and test the CPU on;y version

docker build --build-arg --target dev -t boom-dev --no-cache .
docker compose --profile api up -d

docker run --rm -it  --network boom_boom   -v $(pwd)/config.docker.yaml:/app/config.yaml   -v $(pwd)/data:/app/data   -v $(pwd)/.env:/app/.env   boom-dev bash

Inside bash

cargo run --release  --bin kafka_producer ztf 20240617 public --server-url broker:29092

cargo run --release --bin kafka_consumer ztf 20240617 --programids public

cargo run --release  --bin scheduler ztf --config config.yaml

Example of an output

[frenbox]

[frenbox]

Another PR was created (#448). Closing this.