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

Commit 935cc9e

Browse files
committed
Use _ThreadLocalState.useLazyTensor in tests.
1 parent e63aa70 commit 935cc9e

File tree

4 files changed

+8
-8
lines changed

4 files changed

+8
-8
lines changed

Tests/TensorFlowTests/LazyTensorEvaluationTests.swift

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,12 +20,12 @@ import CTensorFlow
2020
final class LazyTensorEvaluationTests: XCTestCase {
2121
override class func setUp() {
2222
super.setUp()
23-
_RuntimeConfig.useLazyTensor = true
23+
_ThreadLocalState.useLazyTensor = true
2424
}
2525

2626
override class func tearDown() {
2727
super.tearDown()
28-
_RuntimeConfig.useLazyTensor = false
28+
_ThreadLocalState.useLazyTensor = false
2929
}
3030

3131
func testSimpleOperations() {

Tests/TensorFlowTests/LazyTensorExplicitTraceTests.swift

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,12 +20,12 @@ import CTensorFlow
2020
final class LazyTensorExplicitTraceTests: XCTestCase {
2121
override class func setUp() {
2222
super.setUp()
23-
_RuntimeConfig.useLazyTensor = true
23+
_ThreadLocalState.useLazyTensor = true
2424
}
2525

2626
override class func tearDown() {
2727
super.tearDown()
28-
_RuntimeConfig.useLazyTensor = false
28+
_ThreadLocalState.useLazyTensor = false
2929
}
3030

3131
func testSingleInput() {

Tests/TensorFlowTests/LazyTensorTFFunctionBuilderTests.swift

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,12 @@ import CTensorFlow
1919
final class LazyTensorTFFunctionBuilderTests : XCTestCase {
2020
override class func setUp() {
2121
super.setUp()
22-
_RuntimeConfig.useLazyTensor = true
22+
_ThreadLocalState.useLazyTensor = true
2323
}
2424

2525
override class func tearDown() {
2626
super.tearDown()
27-
_RuntimeConfig.useLazyTensor = false
27+
_ThreadLocalState.useLazyTensor = false
2828
}
2929

3030
func testSingletonInputs() {

Tests/TensorFlowTests/LazyTensorTraceTests.swift

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,12 +20,12 @@ import CTensorFlow
2020
final class LazyTensorTraceTests: XCTestCase {
2121
override class func setUp() {
2222
super.setUp()
23-
_RuntimeConfig.useLazyTensor = true
23+
_ThreadLocalState.useLazyTensor = true
2424
}
2525

2626
override class func tearDown() {
2727
super.tearDown()
28-
_RuntimeConfig.useLazyTensor = false
28+
_ThreadLocalState.useLazyTensor = false
2929
}
3030

3131
func testSingleLiveTensor() {

0 commit comments

Comments
 (0)