github.com/apache/arrow/go/v7@v7.0.1/parquet/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  // +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 _int32_max_min_avx2(values unsafe.Pointer, length int, minout, maxout unsafe.Pointer)
    30  
    31  func int32MaxMinAVX2(values []int32) (min, max int32) {
    32  	_int32_max_min_avx2(unsafe.Pointer(&values[0]), len(values), unsafe.Pointer(&min), unsafe.Pointer(&max))
    33  	return
    34  }
    35  
    36  //go:noescape
    37  func _uint32_max_min_avx2(values unsafe.Pointer, length int, minout, maxout unsafe.Pointer)
    38  
    39  func uint32MaxMinAVX2(values []uint32) (min, max uint32) {
    40  	_uint32_max_min_avx2(unsafe.Pointer(&values[0]), len(values), unsafe.Pointer(&min), unsafe.Pointer(&max))
    41  	return
    42  }
    43  
    44  //go:noescape
    45  func _int64_max_min_avx2(values unsafe.Pointer, length int, minout, maxout unsafe.Pointer)
    46  
    47  func int64MaxMinAVX2(values []int64) (min, max int64) {
    48  	_int64_max_min_avx2(unsafe.Pointer(&values[0]), len(values), unsafe.Pointer(&min), unsafe.Pointer(&max))
    49  	return
    50  }
    51  
    52  //go:noescape
    53  func _uint64_max_min_avx2(values unsafe.Pointer, length int, minout, maxout unsafe.Pointer)
    54  
    55  func uint64MaxMinAVX2(values []uint64) (min, max uint64) {
    56  	_uint64_max_min_avx2(unsafe.Pointer(&values[0]), len(values), unsafe.Pointer(&min), unsafe.Pointer(&max))
    57  	return
    58  }