Unpack also works on the GPU.

ledatelescope · Aug 29, 2023 · fe8568c · fe8568c
1 parent ab80ce3
commit fe8568c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/bifrost/blocks/unpack.py b/python/bifrost/blocks/unpack.py
@@ -46,7 +46,7 @@ def __init__(self, iring: Ring, dtype: Union[str,np.dtype], align_msb: bool=Fals
         self.align_msb = align_msb
     def define_valid_input_spaces(self) -> Tuple[str]:
         """Return set of valid spaces (or 'any') for each input"""
-        return ('system',)
+        return ('any',)
     def on_sequence(self, iseq: ReadSequence) -> Dict[str,Any]:
         ihdr = iseq.header
         ohdr = deepcopy(ihdr)
@@ -76,8 +76,8 @@ def unpack(iring: Ring, dtype: Union[str,np.dtype], *args, **kwargs) -> UnpackBl
 
     **Tensor semantics**::
 
-        Input:  [...], dtype = one of: i/u2, i/u4, ci2, ci4, space = SYSTEM
-        Output: [...], dtype = i8 or ci8 (matching input), space = SYSTEM
+        Input:  [...], dtype = one of: i/u2, i/u4, ci2, ci4, space = SYSTEM or CUDA
+        Output: [...], dtype = i8 or ci8 (matching input), space = SYSTEM or CUDA
 
     Returns:
         UnpackBlock: A new block instance.