Skip to content
This repository was archived by the owner on Jul 1, 2023. It is now read-only.

Commit b6a0df7

Browse files
asuhanainu-bot
andauthored
Update x10 CUDA build instructions (#939)
* Update x10 CUDA build instructions * Fix code style issues with swift-format Co-authored-by: Swift for TensorFlow <ainu-bot@google.com>
1 parent 02a4170 commit b6a0df7

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

README.md

+3-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
Get a taste of *protocol-oriented differentiable programming*.
44

5-
This repository hosts [Swift for TensorFlow](https://github.com/tensorflow/swift)'s deep learning library, available both as a part of Swift for TensorFlow toolchains and as a Swift package.
5+
This repository hosts [Swift for TensorFlow](https://github.com/tensorflow/swift)'s deep learning library, available both as a part of Swift for TensorFlow toolchains and as a Swift package.
66

77
## Usage
88

@@ -112,6 +112,8 @@ and its argument are elided).
112112
If `swiftc` is not in your `PATH`, you must specify the path to it using
113113
`-D CMAKE_Swift_COMPILER=`.
114114

115+
To enable CUDA support, run `export TF_NEED_CUDA=1` before the steps below.
116+
115117
```
116118
cmake -G Ninja -B out -S swift-apis
117119
cmake --build out

Tests/TensorFlowTests/EpochsTests.swift

+1-1
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ final class EpochsTests: XCTestCase {
158158
var epochSampleCount = 0
159159
for batch in epochBatches {
160160
XCTAssertEqual(
161-
batch.count, batchSize, "unexpected batch size: \(batch.count)")
161+
batch.count, batchSize, "unexpected batch size: \(batch.count)")
162162
epochSampleCount += batch.count
163163
}
164164
let expectedDropCount = dataset.count % 64

0 commit comments

Comments
 (0)