Skip to content
Snippets Groups Projects
Commit 1527a1c8 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Driver] don't use cacheable for Alveo driver

parent d9e0a722
No related branches found
No related tags found
No related merge requests found
......@@ -150,8 +150,12 @@ class FINNAccelDriver():
raise ValueError("Supported platforms are zynq zynq-iodma alveo")
# allocate a PYNQ buffer for the packed input and buffer
self.ibuf_packed_device = allocate(shape=self.ishape_packed, dtype=np.uint8, cacheable=True)
self.obuf_packed_device = allocate(shape=self.oshape_packed, dtype=np.uint8, cacheable=True)
if self.platform == "alveo":
self.ibuf_packed_device = allocate(shape=self.ishape_packed, dtype=np.uint8)
self.obuf_packed_device = allocate(shape=self.oshape_packed, dtype=np.uint8)
else:
self.ibuf_packed_device = allocate(shape=self.ishape_packed, dtype=np.uint8, cacheable=True)
self.obuf_packed_device = allocate(shape=self.oshape_packed, dtype=np.uint8, cacheable=True)
def fold_input(self, ibuf_normal):
\"\"\"Reshapes input in desired shape.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment