()
| 11 | |
| 12 | # Demonstrate dlpack consumption invalidates it for future use |
| 13 | def dlp_usage_error(): |
| 14 | a = cp.empty((3,3), dtype=cp.float32) |
| 15 | dlp = a.__dlpack__() |
| 16 | assert(matxutil.check_dlpack_status(dlp) == 0) |
| 17 | a2 = cp.from_dlpack(dlp) # causes dlp to become unused |
| 18 | assert(matxutil.check_dlpack_status(dlp) != 0) |
| 19 | return dlp |
| 20 | |
| 21 | # Demonstrate cupy array stays in scope when returning valid dlp |
| 22 | def scope_okay(): |