Files
2026-07-13 12:47:05 +08:00

188 lines
6.9 KiB
JSON

{
"version": 2,
"configurePresets": [
{
"name": "base_cpu_onednn",
"displayName": "Configure preset for the base",
"description": "Sets generator, build and install directory",
"generator": "Unix Makefiles",
"binaryDir": "${sourceDir}/blasbuild/cpu",
"cacheVariables": {
"CMAKE_TOOLCHAIN_FILE": "",
"CMAKE_INSTALL_PREFIX": "${sourceDir}/out/install/${presetName}",
"SD_LIBRARY_NAME": "nd4jcpu",
"SD_CPU": true,
"HELPERS_onednn": "ON",
"GENERATE_FLATC": "ON",
"SD_ARCH": "x86-64",
"SD_GCC_FUNCTRACE": "OFF",
"PRINT_MATH": "ON",
"PRINT_INDICES": "ON",
"SD_ALL_OPS": true,
"CMAKE_BUILD_TYPE" : "Debug",
"OPENBLAS_PATH": "$env{HOME}/.javacpp/cache/openblas-0.3.19-1.5.7-linux-x86_64.jar/org/bytedeco/openblas/linux-x86_64"
}
},
{
"name": "base_cpu",
"displayName": "Configure preset for the base",
"description": "Sets generator, build and install directory",
"generator": "Unix Makefiles",
"binaryDir": "${sourceDir}/blasbuild/cpu",
"cacheVariables": {
"CMAKE_TOOLCHAIN_FILE": "",
"CMAKE_INSTALL_PREFIX": "${sourceDir}/out/install/${presetName}",
"SD_LIBRARY_NAME": "nd4jcpu",
"SD_CPU": true,
"GENERATE_FLATC": "ON",
"SD_ARCH": "x86-64",
"SD_GCC_FUNCTRACE": "OFF",
"PRINT_MATH": "ON",
"PRINT_INDICES": "ON",
"SD_ALL_OPS": true,
"CMAKE_BUILD_TYPE" : "Debug",
"OPENBLAS_PATH": "$env{HOME}/.javacpp/cache/openblas-0.3.19-1.5.7-linux-x86_64.jar/org/bytedeco/openblas/linux-x86_64"
}
},
{
"name": "base_cpu_tests",
"displayName": "Configure preset for the base with tests",
"description": "Sets generator, build and install directory",
"generator": "Unix Makefiles",
"binaryDir": "${sourceDir}/blasbuild/cpu",
"cacheVariables": {
"CMAKE_TOOLCHAIN_FILE": "",
"CMAKE_INSTALL_PREFIX": "${sourceDir}/out/install/${presetName}",
"SD_LIBRARY_NAME": "nd4jcpu",
"SD_CPU": true,
"SD_ARCH": "x86-64",
"PRINT_INDICES": "ON",
"SD_BUILD_TESTS": "OFF",
"PRINT_MATH": "ON",
"GENERATE_FLATC": "ON",
"SD_ALL_OPS": true,
"CMAKE_BUILD_TYPE" : "Debug",
"OPENBLAS_PATH": "$env{HOME}/.javacpp/cache/openblas-0.3.19-1.5.7-linux-x86_64.jar/org/bytedeco/openblas/linux-x86_64"
}
},
{
"name": "base_cuda",
"displayName": "Configure preset for the base Cuda",
"description": "Sets generator, build and install directory",
"hidden": true,
"generator": "Unix Makefiles",
"binaryDir": "${sourceDir}/blasbuild/cuda",
"cacheVariables": {
"CMAKE_TOOLCHAIN_FILE": "",
"CMAKE_INSTALL_PREFIX": "${sourceDir}/out/install/${presetName}",
"SD_LIBRARY_NAME": "nd4jcuda",
"SD_ALL_OPS": true,
"SD_CUDA": true,
"BLAS":true,
"CMAKE_BUILD_TYPE" : "Debug",
"COMPUTE": "8.0",
"__CUDACC__" : "ON",
"SD_GCC_FUNCTRACE": "OFF",
"CMAKE_CUDA_ARCHITECTURES": "86",
"SD_BUILD_TESTS": "OFF",
"GENERATE_FLATC": "ON",
"CUDA_TOOLKIT_ROOT_DIR": "/usr/local/cuda",
"CMAKE_CUDA_COMPILER": "/usr/local/cuda/bin/nvcc"
}
},
{
"name": "base_cuda_tests",
"displayName": "Configure preset for the base Cuda",
"description": "Sets generator, build and install directory",
"hidden": true,
"generator": "Unix Makefiles",
"binaryDir": "${sourceDir}/blasbuild/cuda",
"cacheVariables": {
"CMAKE_TOOLCHAIN_FILE": "",
"CMAKE_INSTALL_PREFIX": "${sourceDir}/out/install/${presetName}",
"SD_LIBRARY_NAME": "nd4jcuda",
"SD_ALL_OPS": true,
"SD_CUDA": true,
"GENERATE_FLATC": "ON",
"BLAS":true,
"SD_GCC_FUNCTRACE": "OFF",
"__CUDACC__" : "ON",
"CMAKE_BUILD_TYPE" : "Debug",
"COMPUTE": "86",
"CUDA_TOOLKIT_ROOT_DIR": "/usr/local/cuda-12.1",
"CMAKE_CUDA_COMPILER": "/usr/local/cuda-12.2/bin/nvcc"
}
},
{
"name": "cuda_cudnn",
"displayName": "Configure preset for the CUDA and CUDNN",
"description": "Sets Unix Makefile generator, build and install directory",
"hidden": false,
"inherits": [
"base_cuda"
],
"generator": "Unix Makefiles",
"binaryDir": "${sourceDir}/blasbuild/cuda/${presetName}",
"cacheVariables": {
"SD_ARCH": "x86-64",
"HELPERS_cudnn": true
}
},
{
"name": "cuda_cudnn_debug",
"displayName": "Configure Debug preset for the CUDA and CUDNN",
"description": "Sets Unix Makefile generator, build and install directory",
"hidden": false,
"inherits": [
"cuda_cudnn"
],
"generator": "Unix Makefiles",
"binaryDir": "${sourceDir}/blasbuild/cuda/${presetName}",
"cacheVariables": {
"SD_BUILD_TESTS": "OFF",
"CMAKE_BUILD_TYPE": "Debug"
}
}
],
"buildPresets": [
{
"name": "cpu_build",
"description": "",
"displayName": "",
"configurePreset": "base_cpu",
"jobs": 64
},
{
"name": "cuda_cudnn_debug_build",
"description": "",
"displayName": "",
"configurePreset": "cuda_cudnn_debug",
"jobs": 64
},
{
"name": "cpu_debug_test",
"description": "",
"displayName": "",
"configurePreset": "cuda_cudnn_debug",
"jobs": 64
}
],
"testPresets": [
{
"name": "cuda_cudnn_debug_test",
"description": "",
"displayName": "",
"configurePreset": "cuda_cudnn_debug"
},
{
"name": "cpu_debug_test",
"description": "",
"displayName": "",
"configurePreset": "base_cpu_tests"
}
]
}