MCPcopy Create free account
hub / github.com/ROCm/rocPRIM / SelectionType

Class SelectionType

scripts/autotune/create_optimization.py:67–78  ·  view source on GitHub ↗

Data class describing a type used to select a configuration.

Source from the content-addressed store, hash-verified

65
66@dataclass
67class SelectionType:
68 """
69 Data class describing a type used to select a configuration.
70 """
71 name: str
72 # True if rocprim::empty_type is a valid type in the algorithm's configuration.
73 is_optional: bool
74 # True if only the size of the type is relevant. If false, selection also happens based on rocprim::is_floating.
75 # Only selecting on size will limit the number of tuning benchmarks required and configs generated.
76 # For example, a sort by key will not perform any logic on the value type and only move the data.
77 # Hence it can select the value type on size only.
78 select_on_size_only: bool
79
80@dataclass
81class SelectionConst:

Calls

no outgoing calls

Tested by

no test coverage detected