ceph/src/arrow/go/parquet/internal/utils/min_max_amd64.go

   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 "golang.org/x/sys/cpu"
  22
  23 func init() {
  24         // if the CPU supports AVX2 or SSE4 then let's use those to benefit from SIMD
  25         // to accelerate the performance for finding the min and max for an integral slice.
  26         // otherwise fallback to a pure go implementation if the cpu doesn't have these features.
  27         if cpu.X86.HasAVX2 {
  28                 minmaxFuncs.i32 = int32MaxMinAVX2
  29                 minmaxFuncs.ui32 = uint32MaxMinAVX2
  30                 minmaxFuncs.i64 = int64MaxMinAVX2
  31                 minmaxFuncs.ui64 = uint64MaxMinAVX2
  32         } else if cpu.X86.HasSSE42 {
  33                 minmaxFuncs.i32 = int32MaxMinSSE4
  34                 minmaxFuncs.ui32 = uint32MaxMinSSE4
  35                 minmaxFuncs.i64 = int64MaxMinSSE4
  36                 minmaxFuncs.ui64 = uint64MaxMinSSE4
  37         } else {
  38                 minmaxFuncs.i32 = int32MinMax
  39                 minmaxFuncs.ui32 = uint32MinMax
  40                 minmaxFuncs.i64 = int64MinMax
  41                 minmaxFuncs.ui64 = uint64MinMax
  42         }
  43 }