188 lines
6.9 KiB
JSON
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"
|
|
}
|
|
]
|
|
}
|