This repository was archived by the owner on Jul 1, 2023. It is now read-only.
File tree Expand file tree Collapse file tree 4 files changed +8
-8
lines changed Expand file tree Collapse file tree 4 files changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -20,12 +20,12 @@ import CTensorFlow
20
20
final class LazyTensorEvaluationTests : XCTestCase {
21
21
override class func setUp( ) {
22
22
super. setUp ( )
23
- _RuntimeConfig . useLazyTensor = true
23
+ _ThreadLocalState . useLazyTensor = true
24
24
}
25
25
26
26
override class func tearDown( ) {
27
27
super. tearDown ( )
28
- _RuntimeConfig . useLazyTensor = false
28
+ _ThreadLocalState . useLazyTensor = false
29
29
}
30
30
31
31
func testSimpleOperations( ) {
Original file line number Diff line number Diff line change @@ -20,12 +20,12 @@ import CTensorFlow
20
20
final class LazyTensorExplicitTraceTests : XCTestCase {
21
21
override class func setUp( ) {
22
22
super. setUp ( )
23
- _RuntimeConfig . useLazyTensor = true
23
+ _ThreadLocalState . useLazyTensor = true
24
24
}
25
25
26
26
override class func tearDown( ) {
27
27
super. tearDown ( )
28
- _RuntimeConfig . useLazyTensor = false
28
+ _ThreadLocalState . useLazyTensor = false
29
29
}
30
30
31
31
func testSingleInput( ) {
Original file line number Diff line number Diff line change @@ -19,12 +19,12 @@ import CTensorFlow
19
19
final class LazyTensorTFFunctionBuilderTests : XCTestCase {
20
20
override class func setUp( ) {
21
21
super. setUp ( )
22
- _RuntimeConfig . useLazyTensor = true
22
+ _ThreadLocalState . useLazyTensor = true
23
23
}
24
24
25
25
override class func tearDown( ) {
26
26
super. tearDown ( )
27
- _RuntimeConfig . useLazyTensor = false
27
+ _ThreadLocalState . useLazyTensor = false
28
28
}
29
29
30
30
func testSingletonInputs( ) {
Original file line number Diff line number Diff line change @@ -20,12 +20,12 @@ import CTensorFlow
20
20
final class LazyTensorTraceTests : XCTestCase {
21
21
override class func setUp( ) {
22
22
super. setUp ( )
23
- _RuntimeConfig . useLazyTensor = true
23
+ _ThreadLocalState . useLazyTensor = true
24
24
}
25
25
26
26
override class func tearDown( ) {
27
27
super. tearDown ( )
28
- _RuntimeConfig . useLazyTensor = false
28
+ _ThreadLocalState . useLazyTensor = false
29
29
}
30
30
31
31
func testSingleLiveTensor( ) {
You can’t perform that action at this time.
0 commit comments