Skip to content

Commit

Permalink
Merge pull request #22424 from lexming/20250228154955_new_pr_ollama0512
Browse files Browse the repository at this point in the history
{ai}[GCCcore/13.3.0] ollama v0.5.12, Go v1.23.6 w/ CUDA 12.6.0
  • Loading branch information
smoors authored Mar 3, 2025
2 parents 5054c2e + 8bd1967 commit 746163b
Show file tree
Hide file tree
Showing 3 changed files with 130 additions and 0 deletions.
29 changes: 29 additions & 0 deletions easybuild/easyconfigs/g/Go/Go-1.23.6.eb
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
easyblock = 'Tarball'

name = 'Go'
version = '1.23.6'

homepage = 'https://www.golang.org'
description = """Go is an open source programming language that makes it easy to build
simple, reliable, and efficient software."""

toolchain = SYSTEM

source_urls = ['https://storage.googleapis.com/golang/']
local_archs = {'aarch64': 'arm64', 'x86_64': 'amd64'}
sources = ['go%%(version)s.linux-%s.tar.gz' % local_archs[ARCH]]
checksums = [{
'go%(version)s.linux-amd64.tar.gz': '9379441ea310de000f33a4dc767bd966e72ab2826270e038e78b2c53c2e7802d',
'go%(version)s.linux-arm64.tar.gz': '561c780e8f4a8955d32bf72e46af0b5ee5e0debe1e4633df9a03781878219202',
}]

sanity_check_paths = {
'files': ['bin/go', 'bin/gofmt'],
'dirs': ['api', 'doc', 'lib', 'pkg'],
}

sanity_check_commands = ["go help"]

modextravars = {'GOROOT': '%(installdir)s'}

moduleclass = 'compiler'
Original file line number Diff line number Diff line change
@@ -0,0 +1,55 @@
easyblock = 'Bundle'

name = 'ollama'
version = '0.5.12'
versionsuffix = '-CUDA-%(cudaver)s'

homepage = 'https://ollama.com/'
description = "Get up and running with large language models."

toolchain = {'name': 'GCCcore', 'version': '13.3.0'}

builddependencies = [
('Go', '1.23.6', '', SYSTEM),
('binutils', '2.42'),
('CMake', '3.29.3'),
]

dependencies = [
('CUDA', '12.6.0', '', SYSTEM),
]

# default CUDA compute capabilities to use (override via --cuda-compute-capabilities)
cuda_compute_capabilities = ['6.0', '7.0', '7.5', '8.0', '8.6', '9.0']

default_component_specs = {
'source_urls': ['https://github.com/ollama/ollama/archive'],
'sources': ['v%(version)s.tar.gz'],
'checksums': ['a38d5e3ce4ee13d237ef670231016d02ee81cb880f38d58795e989c73b5b7523'],
'start_dir': 'ollama-%(version)s',
}
components = [
('ggml-cpu-cuda', version, {
# CPU and CUDA12 presets
'easyblock': 'CMakeMake',
'buildopts': 'ggml-cpu ggml-cuda',
}),
('ollama', version, {
'easyblock': 'GoPackage',
}),
]

sanity_check_paths = {
'files': ['bin/ollama', f'lib/ollama/libggml-base.{SHLIB_EXT}', f'lib/ollama/cuda_v12/libggml-cuda.{SHLIB_EXT}'],
'dirs': ['lib/ollama', 'lib/ollama/cuda_v12'],
}

sanity_check_commands = [
"ollama -v",
]

modextrapaths = {
'LD_LIBRARY_PATH': ['lib/ollama', 'lib/ollama/cuda_v12'],
}

moduleclass = 'ai'
46 changes: 46 additions & 0 deletions easybuild/easyconfigs/o/ollama/ollama-0.5.12-GCCcore-13.3.0.eb
Original file line number Diff line number Diff line change
@@ -0,0 +1,46 @@
easyblock = 'Bundle'

name = 'ollama'
version = '0.5.12'

homepage = 'https://ollama.com/'
description = "Get up and running with large language models."

toolchain = {'name': 'GCCcore', 'version': '13.3.0'}

builddependencies = [
('Go', '1.23.6', '', SYSTEM),
('binutils', '2.42'),
('CMake', '3.29.3'),
]

default_component_specs = {
'source_urls': ['https://github.com/ollama/ollama/archive'],
'sources': ['v%(version)s.tar.gz'],
'checksums': ['a38d5e3ce4ee13d237ef670231016d02ee81cb880f38d58795e989c73b5b7523'],
'start_dir': 'ollama-%(version)s',
}
components = [
('ggml-cpu', version, {
'easyblock': 'CMakeMake',
'buildopts': '%(name)s',
}),
('ollama', version, {
'easyblock': 'GoPackage',
}),
]

sanity_check_paths = {
'files': ['bin/ollama', f'lib/ollama/libggml-base.{SHLIB_EXT}'],
'dirs': ['lib/ollama'],
}

sanity_check_commands = [
"ollama -v",
]

modextrapaths = {
'LD_LIBRARY_PATH': 'lib/ollama',
}

moduleclass = 'ai'

0 comments on commit 746163b

Please sign in to comment.