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

Function create_algorithm

scripts/autotune/create_optimization.py:697–768  ·  view source on GitHub ↗
(algorithm_name: str, fallback_entries: List[FallbackCase])

Source from the content-addressed store, hash-verified

695 return True
696
697def create_algorithm(algorithm_name: str, fallback_entries: List[FallbackCase]):
698 fallback_entries = list(filter(lambda e: filt_algo_regex(e, algorithm_name), fallback_entries))
699 if algorithm_name == 'device_histogram':
700 return AlgorithmDeviceHistogram(fallback_entries)
701 elif algorithm_name == 'device_merge_sort_block_sort':
702 return AlgorithmDeviceMergeSortBlockSort(fallback_entries)
703 elif algorithm_name == 'device_merge_sort_block_merge':
704 return AlgorithmDeviceMergeSortBlockMerge(fallback_entries)
705 elif algorithm_name == 'device_radix_sort_block_sort':
706 return AlgorithmDeviceRadixSortBlockSort(fallback_entries)
707 elif algorithm_name == 'device_radix_sort_onesweep':
708 return AlgorithmDeviceRadixSortOnesweep(fallback_entries)
709 elif algorithm_name == 'device_reduce':
710 return AlgorithmDeviceReduce(fallback_entries)
711 elif algorithm_name == 'device_segmented_reduce':
712 return AlgorithmDeviceSegmentedReduce(fallback_entries)
713 elif algorithm_name == 'device_scan':
714 return AlgorithmDeviceScan(fallback_entries)
715 elif algorithm_name == 'device_scan_by_key':
716 return AlgorithmDeviceScanByKey(fallback_entries)
717 elif algorithm_name == 'device_binary_search':
718 return AlgorithmDeviceBinarySearch(fallback_entries)
719 elif algorithm_name == 'device_upper_bound':
720 return AlgorithmDeviceUpperBound(fallback_entries)
721 elif algorithm_name == 'device_lower_bound':
722 return AlgorithmDeviceLowerBound(fallback_entries)
723 elif algorithm_name == 'device_adjacent_difference':
724 return AlgorithmDeviceAdjacentDifference(fallback_entries)
725 elif algorithm_name == 'device_adjacent_difference_inplace':
726 return AlgorithmDeviceAdjacentDifferenceInplace(fallback_entries)
727 elif algorithm_name == 'device_adjacent_find':
728 return AlgorithmDeviceAdjacentFind(fallback_entries)
729 elif algorithm_name == 'device_segmented_radix_sort':
730 return AlgorithmDeviceSegmentedRadixSort(fallback_entries)
731 elif algorithm_name == 'device_transform':
732 return AlgorithmDeviceTransform(fallback_entries)
733 elif algorithm_name == 'device_transform_pointer':
734 return AlgorithmDeviceTransformPointer(fallback_entries)
735 elif algorithm_name == 'device_partition_two_way_predicate':
736 return AlgorithmDevicePartitionTwoWayPredicate(fallback_entries)
737 elif algorithm_name == 'device_partition_two_way_flag':
738 return AlgorithmDevicePartitionTwoWayFlag(fallback_entries)
739 elif algorithm_name == 'device_partition_flag':
740 return AlgorithmDevicePartitionFlag(fallback_entries)
741 elif algorithm_name == 'device_partition_predicate':
742 return AlgorithmDevicePartitionPredicate(fallback_entries)
743 elif algorithm_name == 'device_partition_three_way':
744 return AlgorithmDevicePartitionThreeWay(fallback_entries)
745 elif algorithm_name == 'device_search_n':
746 return AlgorithmDeviceSearchN(fallback_entries)
747 elif algorithm_name == 'device_select_flag':
748 return AlgorithmDeviceSelectFlag(fallback_entries)
749 elif algorithm_name == 'device_select_predicate':
750 return AlgorithmDeviceSelectPredicate(fallback_entries)
751 elif algorithm_name == 'device_select_predicated_flag':
752 return AlgorithmDeviceSelectPredicatedFlag(fallback_entries)
753 elif algorithm_name == 'device_select_unique':
754 return AlgorithmDeviceSelectUnique(fallback_entries)

Callers 1

Tested by

no test coverage detected