diff --git a/Setup.hs b/Setup.hs index 9e7bf6b..b08c1aa 100644 --- a/Setup.hs +++ b/Setup.hs @@ -223,11 +223,12 @@ cudaLibraryPaths (Platform arch os) installPath = [ installPath path | path where libpaths = case (os, arch) of - (Windows, I386) -> ["lib/Win32"] - (Windows, X86_64) -> ["lib/x64"] - (OSX, _) -> ["lib"] -- MacOS does not distinguish 32- vs. 64-bit paths - (_, X86_64) -> ["lib64", "lib"] -- prefer lib64 for 64-bit systems - _ -> ["lib"] + (Windows, I386) -> ["lib/Win32"] + (Windows, X86_64) -> ["lib/x64"] + (OSX, _) -> ["lib"] -- MacOS does not distinguish 32- vs. 64-bit paths + (_, X86_64) -> ["lib64", "lib"] -- prefer lib64 for 64-bit systems + (_, AArch64) -> ["lib64", "lib"] + _ -> ["lib"] -- otherwise -- On Windows and OSX we use different libraries depending on whether we are