-
Notifications
You must be signed in to change notification settings - Fork 0
/
Package.swift
90 lines (88 loc) · 2.72 KB
/
Package.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
// swift-tools-version: 5.8
// The swift-tools-version declares the minimum version of Swift required to build this package.
import PackageDescription
let package = Package(
name: "Sggml",
platforms: [
.macOS(.v13)
],
products: [
.executable(
name: "mnist",
targets: ["Mnist"]
),
.executable(
name: "gpt2",
targets: ["Gpt2"]
),
.executable(
name: "bert-spam",
targets: ["BertSpam"]
),
.library(
name: "Sggml",
targets: ["Sggml"]
)
],
dependencies: [
.package(url: "https://github.com/apple/swift-log.git", from: "1.4.0"),
.package(url: "https://github.com/apple/swift-argument-parser.git", from: "1.2.0"),
.package(url: "https://github.com/jph00/BaseMath.git", branch: "master"),
.package(url: "https://github.com/huggingface/swift-transformers", from: "0.1.0")
],
targets: [
.target(
name: "Utils"
),
.target(
name: "ggml",
cSettings: [
.unsafeFlags(["-Wno-shorten-64-to-32", "-Ofast"], .when(configuration: .release)),
.unsafeFlags(["-Wno-shorten-64-to-32"], .when(configuration: .debug)),
.define("GGML_USE_ACCELERATE", .when(platforms: [.macOS]))
],
linkerSettings: [
.linkedFramework("Accelerate", .when(platforms: [.macOS]))
]
),
.target(
name: "Sggml",
dependencies: [
"ggml"
]
),
.executableTarget(
name: "BertSpam",
dependencies: [
"Sggml",
"Utils",
.product(name: "ArgumentParser", package: "swift-argument-parser"),
.product(name: "Logging", package: "swift-log"),
.product(name: "Transformers", package: "swift-transformers")
],
exclude: [
"convert.ipynb",
"train.ipynb"
]
),
.executableTarget(
name: "Mnist",
dependencies: [
"Sggml",
"Utils",
.product(name: "ArgumentParser", package: "swift-argument-parser"),
.product(name: "Logging", package: "swift-log")
]
),
.executableTarget(
name: "Gpt2",
dependencies: [
"Sggml",
"Utils",
"BaseMath",
.product(name: "ArgumentParser", package: "swift-argument-parser"),
.product(name: "Logging", package: "swift-log")
]
)
]
)