Skip to content
Snippets Groups Projects
Commit 825c6991 authored by Michael Kuron's avatar Michael Kuron :mortar_board: Committed by Markus Holzer
Browse files

Apply better type annotation

parent 0cd17895
No related branches found
No related tags found
1 merge request!341Refactor gpu indexing
...@@ -138,7 +138,7 @@ class BlockIndexing(AbstractIndexing): ...@@ -138,7 +138,7 @@ class BlockIndexing(AbstractIndexing):
device_number: device number of the used GPU. By default, the zeroth device is used. device_number: device number of the used GPU. By default, the zeroth device is used.
""" """
def __init__(self, iteration_space: Tuple[slice], data_layout: Tuple, def __init__(self, iteration_space: Tuple[slice], data_layout: Tuple[int],
block_size=(128, 2, 1), permute_block_size_dependent_on_layout=True, compile_time_block_size=False, block_size=(128, 2, 1), permute_block_size_dependent_on_layout=True, compile_time_block_size=False,
maximum_block_size=(1024, 1024, 64), device_number=None): maximum_block_size=(1024, 1024, 64), device_number=None):
super(BlockIndexing, self).__init__(iteration_space, data_layout) super(BlockIndexing, self).__init__(iteration_space, data_layout)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment