This repository aims to illustrate how to use bazel to build a C++ project with CUDA support.
To build the project, run:
bazel build //src:matmul
To run the project, run:
bazel run //src:matmul
or directly:
./bazel-bin/src/matmul
To build the docker image, run:
cp ./bazel-bin/src/matmul .
DOCKER_BUILDKIT=1 docker build -t machinelearning-one/cubzl:latest .
To run the docker image, run:
docker run --rm -it --gpus all machinelearning-one/cubzl:latest