Skip to content

Commit 455cb96

Browse files
committed
[Driver] Run pre-commit
1 parent 41fc498 commit 455cb96

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

src/finn/qnn-data/templates/driver/driver_base.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -358,12 +358,16 @@ def execute_on_buffers(self, asynch=False, batch_size=None):
358358
iwdma.write(0x00, 1)
359359
for o in range(self.num_outputs):
360360
self.odma[o].write(0x10, self.obuf_packed_device[o].device_address & 0xFFFFFFFF)
361-
self.odma[o].write(0x14, (self.obuf_packed_device[o].device_address >> 32) & 0xFFFFFFFF)
361+
self.odma[o].write(
362+
0x14, (self.obuf_packed_device[o].device_address >> 32) & 0xFFFFFFFF
363+
)
362364
self.odma[o].write(0x1C, batch_size)
363365
self.odma[o].write(0x00, 1)
364366
for i in range(self.num_inputs):
365367
self.idma[i].write(0x10, self.ibuf_packed_device[i].device_address & 0xFFFFFFFF)
366-
self.idma[i].write(0x14, (self.ibuf_packed_device[i].device_address >> 32) & 0xFFFFFFFF)
368+
self.idma[i].write(
369+
0x14, (self.ibuf_packed_device[i].device_address >> 32) & 0xFFFFFFFF
370+
)
367371
self.idma[i].write(0x1C, batch_size)
368372
self.idma[i].write(0x00, 1)
369373
elif self.platform == "alveo":

0 commit comments

Comments
 (0)