File tree 1 file changed +17
-9
lines changed
1 file changed +17
-9
lines changed Original file line number Diff line number Diff line change @@ -246,15 +246,23 @@ def eval():
246
246
info (f"Median rate: { perf_rate :.5f} Mopts/s" )
247
247
248
248
# verify
249
- call , put = args [- 2 ], args [- 1 ]
250
- expected_call = 16.976097804669887
251
- expected_put = 0.34645174725098116
252
- call_value = float (call [0 ])
253
- put_value = float (put [0 ])
254
- assert numpy .allclose (call_value , expected_call )
255
- assert numpy .allclose (put_value , expected_put )
256
-
257
- info ("SUCCESS" )
249
+ if device :
250
+ # FIXME gpu.memcpy to host requires identity layout
251
+ # FIXME reduction on gpu
252
+ # call = args[-2].to_device()
253
+ # put = args[-1].to_device()
254
+ pass
255
+ else :
256
+ call = args [- 2 ]
257
+ put = args [- 1 ]
258
+ expected_call = 16.976097804669887
259
+ expected_put = 0.34645174725098116
260
+ call_value = float (call [0 ])
261
+ put_value = float (put [0 ])
262
+ assert numpy .allclose (call_value , expected_call )
263
+ assert numpy .allclose (put_value , expected_put )
264
+ info ("SUCCESS" )
265
+
258
266
fini ()
259
267
260
268
You can’t perform that action at this time.
0 commit comments