Skip to content

Commit

Permalink
[NFC] Fix test_create_memref.py formatting. (#130)
Browse files Browse the repository at this point in the history
This PR formats `test_create_memref.py` using black formatter.
  • Loading branch information
shelkesagar29 authored Aug 20, 2024
1 parent 7c5bcbf commit b4491f7
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -457,6 +457,7 @@ def create_memref_from_dlpack(arr, module):
print(f"-- Memref dtype: {memref.dtype}")
print(f"-- {module.__name__}.from_dlpack(): {module.from_dlpack(memref)}")


print(f"Test np.array -> client.create_memref_from_dlpack")
create_memref_from_dlpack(np.array([1, 2, 3, 4], dtype=np.int32), np)
create_memref_from_dlpack(np.ones((1, 2, 3), dtype=np.float32), np)
Expand Down Expand Up @@ -512,4 +513,4 @@ def create_dangling_memref():

# CHECK-LABEL: Test memref maintains data's lifetime
# CHECK-NEXT: -- Inner scope: np.from_dlpack(): [1 2]
# CHECK-NEXT: -- Outer scope: np.from_dlpack(): [1 2]
# CHECK-NEXT: -- Outer scope: np.from_dlpack(): [1 2]

0 comments on commit b4491f7

Please sign in to comment.