|
|
|
@ -25,7 +25,8 @@ package("libtorch") |
|
|
|
|
add_configs("python", {description = "Build python interface.", default = false, type = "boolean"}) |
|
|
|
|
add_configs("openmp", {description = "Use OpenMP for parallel code.", default = true, type = "boolean"}) |
|
|
|
|
add_configs("cuda", {description = "Enable CUDA support.", default = false, type = "boolean"}) |
|
|
|
|
add_configs("ninja", {description = "Use ninja as build tool.", default = false, type = "boolean"}) |
|
|
|
|
-- https://github.com/pytorch/pytorch/issues/24186 only ninja is supported on windows |
|
|
|
|
add_configs("ninja", {description = "Use ninja as build tool.", default = is_plat("windows"), type = "boolean"}) |
|
|
|
|
add_configs("blas", {description = "Set BLAS vendor.", default = "openblas", type = "string", values = {"mkl", "openblas", "eigen"}}) |
|
|
|
|
add_configs("pybind11", {description = "Use pybind11 from xrepo.", default = false, type = "boolean"}) |
|
|
|
|
add_configs("protobuf-cpp", {description = "Use protobuf from xrepo.", default = false, type = "boolean"}) |
|
|
|
|