Skip to content
Snippets Groups Projects
Commit 4728c8e7 authored by Richard Angersbach's avatar Richard Angersbach
Browse files

Use default block size if block size fitting fails

parent f77c9c28
No related branches found
No related tags found
1 merge request!454Optimization for GPU block size determination
Pipeline #75657 failed
...@@ -4,6 +4,7 @@ from abc import ABC, abstractmethod ...@@ -4,6 +4,7 @@ from abc import ABC, abstractmethod
from dataclasses import dataclass from dataclasses import dataclass
from typing import cast, Any, Callable from typing import cast, Any, Callable
from itertools import chain from itertools import chain
from warnings import warn
from .functions import Lambda from .functions import Lambda
from .parameters import Parameter from .parameters import Parameter
...@@ -295,10 +296,15 @@ class DynamicBlockSizeLaunchConfiguration(GpuLaunchConfiguration): ...@@ -295,10 +296,15 @@ class DynamicBlockSizeLaunchConfiguration(GpuLaunchConfiguration):
num_work_items = cast( num_work_items = cast(
dim3, tuple(int(wit(**kwargs)) for wit in self._num_work_items) dim3, tuple(int(wit(**kwargs)) for wit in self._num_work_items)
) )
try:
reduced_block_size = cast( reduced_block_size = cast(
dim3, dim3,
self.fit_block_size(num_work_items, self._block_size, self._hw_props), self.fit_block_size(num_work_items, self._block_size, self._hw_props),
) )
except CodegenError:
warn(f"Block size fitting could not determine optimal block size configuration. "
f"Defaulting back to {self._block_size}")
reduced_block_size = self._block_size
grid_size = cast( grid_size = cast(
dim3, dim3,
tuple( tuple(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment