github.com/apache/arrow/go/v14@v14.0.1/internal/utils/min_max_avx2_amd64.go (about)

     1  // Licensed to the Apache Software Foundation (ASF) under one
     2  // or more contributor license agreements.  See the NOTICE file
     3  // distributed with this work for additional information
     4  // regarding copyright ownership.  The ASF licenses this file
     5  // to you under the Apache License, Version 2.0 (the
     6  // "License"); you may not use this file except in compliance
     7  // with the License.  You may obtain a copy of the License at
     8  //
     9  // http://www.apache.org/licenses/LICENSE-2.0
    10  //
    11  // Unless required by applicable law or agreed to in writing, software
    12  // distributed under the License is distributed on an "AS IS" BASIS,
    13  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14  // See the License for the specific language governing permissions and
    15  // limitations under the License.
    16  
    17  //go:build !noasm
    18  
    19  package utils
    20  
    21  import (
    22  	"unsafe"
    23  )
    24  
    25  // This file contains convenience functions for utilizing AVX2 intrinsics to quickly
    26  // and efficiently get the min and max from an integral slice.
    27  
    28  //go:noescape
    29  func _int8_max_min_avx2(values unsafe.Pointer, length int, minout, maxout unsafe.Pointer)
    30  
    31  func int8MaxMinAVX2(values []int8) (min, max int8) {
    32  	_int8_max_min_avx2(unsafe.Pointer(&values[0]), len(values), unsafe.Pointer(&min), unsafe.Pointer(&max))
    33  	return
    34  }
    35  
    36  //go:noescape
    37  func _uint8_max_min_avx2(values unsafe.Pointer, length int, minout, maxout unsafe.Pointer)
    38  
    39  func uint8MaxMinAVX2(values []uint8) (min, max uint8) {
    40  	_uint8_max_min_avx2(unsafe.Pointer(&values[0]), len(values), unsafe.Pointer(&min), unsafe.Pointer(&max))
    41  	return
    42  }
    43  
    44  //go:noescape
    45  func _int16_max_min_avx2(values unsafe.Pointer, length int, minout, maxout unsafe.Pointer)
    46  
    47  func int16MaxMinAVX2(values []int16) (min, max int16) {
    48  	_int16_max_min_avx2(unsafe.Pointer(&values[0]), len(values), unsafe.Pointer(&min), unsafe.Pointer(&max))
    49  	return
    50  }
    51  
    52  //go:noescape
    53  func _uint16_max_min_avx2(values unsafe.Pointer, length int, minout, maxout unsafe.Pointer)
    54  
    55  func uint16MaxMinAVX2(values []uint16) (min, max uint16) {
    56  	_uint16_max_min_avx2(unsafe.Pointer(&values[0]), len(values), unsafe.Pointer(&min), unsafe.Pointer(&max))
    57  	return
    58  }
    59  
    60  //go:noescape
    61  func _int32_max_min_avx2(values unsafe.Pointer, length int, minout, maxout unsafe.Pointer)
    62  
    63  func int32MaxMinAVX2(values []int32) (min, max int32) {
    64  	_int32_max_min_avx2(unsafe.Pointer(&values[0]), len(values), unsafe.Pointer(&min), unsafe.Pointer(&max))
    65  	return
    66  }
    67  
    68  //go:noescape
    69  func _uint32_max_min_avx2(values unsafe.Pointer, length int, minout, maxout unsafe.Pointer)
    70  
    71  func uint32MaxMinAVX2(values []uint32) (min, max uint32) {
    72  	_uint32_max_min_avx2(unsafe.Pointer(&values[0]), len(values), unsafe.Pointer(&min), unsafe.Pointer(&max))
    73  	return
    74  }
    75  
    76  //go:noescape
    77  func _int64_max_min_avx2(values unsafe.Pointer, length int, minout, maxout unsafe.Pointer)
    78  
    79  func int64MaxMinAVX2(values []int64) (min, max int64) {
    80  	_int64_max_min_avx2(unsafe.Pointer(&values[0]), len(values), unsafe.Pointer(&min), unsafe.Pointer(&max))
    81  	return
    82  }
    83  
    84  //go:noescape
    85  func _uint64_max_min_avx2(values unsafe.Pointer, length int, minout, maxout unsafe.Pointer)
    86  
    87  func uint64MaxMinAVX2(values []uint64) (min, max uint64) {
    88  	_uint64_max_min_avx2(unsafe.Pointer(&values[0]), len(values), unsafe.Pointer(&min), unsafe.Pointer(&max))
    89  	return
    90  }