@@ -130,6 +130,8 @@ TensorTests.testAllBackends("ElementIndexing") {
130
130
expectEqual ( [ 43 ] , array0D. scalars)
131
131
}
132
132
133
+ #if !CUDA
134
+ // TODO(https://bugs.swift.org/browse/TF-469): This test fails in GPU mode.
133
135
TensorTests . testAllBackends ( " ElementIndexingAssignment " ) {
134
136
// NOTE: cannot test multiple `Tensor.shape` or `Tensor.scalars` directly
135
137
// until send and receive are implemented (without writing a bunch of mini
@@ -157,6 +159,7 @@ TensorTests.testAllBackends("ElementIndexingAssignment") {
157
159
expectEqual ( Array ( stride ( from: 35.0 , to: 40 , by: 1 ) ) , array1D. scalars)
158
160
expectEqual ( [ 23 ] , array0D. scalars)
159
161
}
162
+ #endif // !CUDA
160
163
161
164
TensorTests . testAllBackends ( " NestedElementIndexing " ) {
162
165
// NOTE: This test could use a clearer name, along with other "indexing"
@@ -205,6 +208,9 @@ TensorTests.testAllBackends("SliceIndexing") {
205
208
expectEqual ( Array ( stride ( from: 3.0 , to: 5 , by: 1 ) ) , array1D. scalars)
206
209
}
207
210
211
+
212
+ #if !CUDA
213
+ // TODO(https://bugs.swift.org/browse/TF-469): This test fails in GPU mode.
208
214
TensorTests . testAllBackends ( " SliceIndexingAssignment " ) {
209
215
// NOTE: cannot test `Tensor.shape` or `Tensor.scalars` directly until send
210
216
// and receive are implemented (without writing a bunch of mini tests).
@@ -232,7 +238,10 @@ TensorTests.testAllBackends("SliceIndexingAssignment") {
232
238
expectEqual ( Array ( stride ( from: 20.0 , to: 30 , by: 1 ) ) , array2D. scalars)
233
239
expectEqual ( Array ( stride ( from: 3.0 , to: 5 , by: 1 ) ) , array1D. scalars)
234
240
}
241
+ #endif // !CUDA
235
242
243
+ #if !CUDA
244
+ // TODO(https://bugs.swift.org/browse/TF-469): This test fails in GPU mode.
236
245
TensorTests . testAllBackends ( " EllipsisIndexing " ) {
237
246
// NOTE: cannot test `Tensor.shape` or `Tensor.scalars` directly until send
238
247
// and receive are implemented (without writing a bunch of mini tests).
@@ -260,6 +269,7 @@ TensorTests.testAllBackends("EllipsisIndexing") {
260
269
expectEqual ( Array ( stride ( from: 20.0 , to: 30 , by: 1 ) ) , array2D. scalars)
261
270
expectEqual ( Array ( stride ( from: 3.0 , to: 5 , by: 1 ) ) , array1D. scalars)
262
271
}
272
+ #endif // !CUDA
263
273
264
274
TensorTests . testAllBackends ( " NewAxisIndexing " ) {
265
275
// NOTE: cannot test `Tensor.shape` or `Tensor.scalars` directly until send
@@ -347,6 +357,8 @@ TensorTests.testAllBackends("StridedSliceIndexing") {
347
357
expectEqual ( Array ( stride ( from: 1.0 , to: 5 , by: 2 ) ) , array1D. scalars)
348
358
}
349
359
360
+ #if !CUDA
361
+ // TODO(https://bugs.swift.org/browse/TF-469): This test fails in GPU mode.
350
362
TensorTests . testAllBackends ( " StridedSliceIndexingAssignment " ) {
351
363
// NOTE: cannot test `Tensor.shape` or `Tensor.scalars` directly until send
352
364
// and receive are implemented (without writing a bunch of mini tests).
@@ -378,6 +390,7 @@ TensorTests.testAllBackends("StridedSliceIndexingAssignment") {
378
390
expectEqual ( Array ( stride ( from: 20.0 , to: 30 , by: 1 ) ) , array2D. scalars)
379
391
expectEqual ( Array ( stride ( from: 3.0 , to: 5 , by: 1 ) ) , array1D. scalars)
380
392
}
393
+ #endif // !CUDA
381
394
382
395
TensorTests . test ( " WholeTensorSlicing " ) {
383
396
let t : Tensor < Int32 > = [ [ [ 1 , 1 , 1 ] , [ 2 , 2 , 2 ] ] ,
0 commit comments