File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -15,12 +15,18 @@ import {getPlatform} from "./getPlatform.js";
1515 * as some of them are inadvisable for the current machine (like CUDA on an x64 Mac machine).
1616 */
1717export async function getLlamaGpuTypes ( include : "supported" | "allValid" ) : Promise < LlamaGpuType [ ] > {
18- if ( include === "supported" )
19- return await getGpuTypesToUseForOption ( "auto" ) ;
20-
2118 const platform = getPlatform ( ) ;
2219 const arch = process . arch ;
2320
21+ if ( include === "supported" ) {
22+ const gpuTypes = new Set ( await getGpuTypesToUseForOption ( "auto" ) ) ;
23+
24+ if ( platform === "win" && arch !== "x64" )
25+ gpuTypes . delete ( "vulkan" ) ; // no Vulkan prebuilt binary yet due to incomplete support for arm64
26+
27+ return [ ...gpuTypes ] ;
28+ }
29+
2430 const res : LlamaGpuType [ ] = [ ] ;
2531
2632 // Metal is not properly supported by llama.cpp on x64 Mac machines
You can’t perform that action at this time.
0 commit comments