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

Function create_algorithm

tuning/configs_header_generation.py:902–971  ·  view source on GitHub ↗
(algorithm_name: str, fallback_entries: List[FallbackCase])

Source from the content-addressed store, hash-verified

900
901
902def create_algorithm(algorithm_name: str, fallback_entries: List[FallbackCase]):
903 fallback_entries = list(
904 filter(lambda e: filt_algo_regex(e, algorithm_name), fallback_entries)
905 )
906 if algorithm_name == "device_histogram":
907 return AlgorithmDeviceHistogram(fallback_entries)
908 elif algorithm_name == "device_merge_sort_block_sort":
909 return AlgorithmDeviceMergeSortBlockSort(fallback_entries)
910 elif algorithm_name == "device_merge_sort_block_merge":
911 return AlgorithmDeviceMergeSortBlockMerge(fallback_entries)
912 elif algorithm_name == "device_radix_sort_block_sort":
913 return AlgorithmDeviceRadixSortBlockSort(fallback_entries)
914 elif algorithm_name == "device_radix_sort_onesweep":
915 return AlgorithmDeviceRadixSortOnesweep(fallback_entries)
916 elif algorithm_name == "device_reduce":
917 return AlgorithmDeviceReduce(fallback_entries)
918 elif algorithm_name == "device_scan":
919 return AlgorithmDeviceScan(fallback_entries)
920 elif algorithm_name == "device_scan_by_key":
921 return AlgorithmDeviceScanByKey(fallback_entries)
922 elif algorithm_name == "device_binary_search":
923 return AlgorithmDeviceBinarySearch(fallback_entries)
924 elif algorithm_name == "device_upper_bound":
925 return AlgorithmDeviceUpperBound(fallback_entries)
926 elif algorithm_name == "device_lower_bound":
927 return AlgorithmDeviceLowerBound(fallback_entries)
928 elif algorithm_name == "device_adjacent_difference":
929 return AlgorithmDeviceAdjacentDifference(fallback_entries)
930 elif algorithm_name == "device_adjacent_difference_inplace":
931 return AlgorithmDeviceAdjacentDifferenceInplace(fallback_entries)
932 elif algorithm_name == "device_adjacent_find":
933 return AlgorithmDeviceAdjacentFind(fallback_entries)
934 elif algorithm_name == "device_segmented_radix_sort":
935 return AlgorithmDeviceSegmentedRadixSort(fallback_entries)
936 elif algorithm_name == "device_transform":
937 return AlgorithmDeviceTransform(fallback_entries)
938 elif algorithm_name == "device_partition_two_way_predicate":
939 return AlgorithmDevicePartitionTwoWayPredicate(fallback_entries)
940 elif algorithm_name == "device_partition_two_way_flag":
941 return AlgorithmDevicePartitionTwoWayFlag(fallback_entries)
942 elif algorithm_name == "device_partition_flag":
943 return AlgorithmDevicePartitionFlag(fallback_entries)
944 elif algorithm_name == "device_partition_predicate":
945 return AlgorithmDevicePartitionPredicate(fallback_entries)
946 elif algorithm_name == "device_partition_three_way":
947 return AlgorithmDevicePartitionThreeWay(fallback_entries)
948 elif algorithm_name == "device_select_flag":
949 return AlgorithmDeviceSelectFlag(fallback_entries)
950 elif algorithm_name == "device_select_predicate":
951 return AlgorithmDeviceSelectPredicate(fallback_entries)
952 elif algorithm_name == "device_select_predicated_flag":
953 return AlgorithmDeviceSelectPredicatedFlag(fallback_entries)
954 elif algorithm_name == "device_select_unique":
955 return AlgorithmDeviceSelectUnique(fallback_entries)
956 elif algorithm_name == "device_select_unique_by_key":
957 return AlgorithmDeviceSelectUniqueByKey(fallback_entries)
958 elif algorithm_name == "device_reduce_by_key":
959 return AlgorithmDeviceReduceByKey(fallback_entries)

Callers 1

add_runMethod · 0.70

Tested by

no test coverage detected