mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-12-29 09:04:17 +00:00
14 lines
741 B
Diff
14 lines
741 B
Diff
diff --git a/llama/make/Makefile.rocm b/llama/make/Makefile.rocm
|
|
index 4ab176b4..cd8be223 100644
|
|
--- a/llama/make/Makefile.rocm
|
|
+++ b/llama/make/Makefile.rocm
|
|
@@ -15,7 +15,7 @@ ifeq ($(OS),windows)
|
|
GPU_COMPILER:=$(GPU_COMPILER_WIN)
|
|
else ifeq ($(OS),linux)
|
|
GPU_LIB_DIR_LINUX := $(HIP_PATH)/lib
|
|
- GPU_COMPILER_LINUX := $(shell X=$$(which hipcc 2>/dev/null) && echo $$X)
|
|
+ GPU_COMPILER_LINUX := $(HIP_PATH)/bin/hipcc
|
|
GPU_COMPILER:=$(GPU_COMPILER_LINUX)
|
|
ROCM_TRANSITIVE_LIBS_INITIAL = $(sort $(shell ldd $(GPU_LIBS) | grep "=>" | cut -f2 -d= | cut -f2 -d' ' | grep -e rocm -e amdgpu -e libtinfo -e libnuma -e libelf))
|
|
GPU_TRANSITIVE_LIBS = $(sort $(shell readlink -f $(ROCM_TRANSITIVE_LIBS_INITIAL)) $(ROCM_TRANSITIVE_LIBS_INITIAL))
|