Address review feedback.

rapidsai · Jul 27, 2023 · 18663c5 · 18663c5
1 parent 88450f4
commit 18663c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/kvikio/_lib/libnvcomp_ll.pyx b/python/kvikio/_lib/libnvcomp_ll.pyx
@@ -387,7 +387,7 @@ class nvCompBatchAlgorithmLZ4(nvCompBatchAlgorithm):
 
     options: nvcompBatchedLZ4Opts_t
 
-    HEADER_SIZE_BYTES: int = 4
+    HEADER_SIZE_BYTES: size_t = sizeof(uint32_t)
 
     def __init__(self, data_type: int = 0, has_header: bool = True):
         """Initialize the codec.
@@ -460,7 +460,7 @@ class nvCompBatchAlgorithmLZ4(nvCompBatchAlgorithm):
                 memcpyAsync(
                     <uintptr_t>comp_chunks[i],
                     psize,
-                    sizeof(uint32_t),
+                    self.HEADER_SIZE_BYTES,
                     cudaMemcpyKind.cudaMemcpyDeviceToDevice,
                     stream.ptr
                 )