github.com/apache/arrow/go/v14@v14.0.2/internal/utils/transpose_ints_amd64.go.tmpl (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  // +build !noasm
    19  
    20  package utils
    21  
    22  import (
    23      "golang.org/x/sys/cpu"
    24  )
    25  
    26  var (
    27  {{ $typelist := .In }}
    28  {{range .In}}
    29  {{ $src := .Type -}}
    30  {{ $srcName := .Name -}}
    31  {{ range $typelist -}}
    32  {{ $dest := .Type -}}
    33  {{ $destName := .Name -}}
    34      Transpose{{$srcName}}{{$destName}} func([]{{$src}}, []{{$dest}}, []int32)
    35  {{end}}
    36  {{end}}
    37  )
    38  
    39  
    40  func init() {
    41      if cpu.X86.HasAVX2 {
    42  {{ $typelist := .In }}
    43  {{range .In}}
    44  {{ $src := .Type -}}
    45  {{ $srcName := .Name -}}
    46  {{ range $typelist -}}
    47  {{ $dest := .Type -}}
    48  {{ $destName := .Name -}}
    49          Transpose{{$srcName}}{{$destName}} = transpose{{ $srcName }}{{ $destName }}avx2
    50  {{end}}
    51  {{end}}
    52      } else if cpu.X86.HasSSE42 {
    53  {{ $typelist := .In }}
    54  {{range .In}}
    55  {{ $src := .Type -}}
    56  {{ $srcName := .Name -}}
    57  {{ range $typelist -}}
    58  {{ $dest := .Type -}}
    59  {{ $destName := .Name -}}
    60          Transpose{{$srcName}}{{$destName}} = transpose{{ $srcName }}{{ $destName }}sse4
    61  {{end}}
    62  {{end}}    
    63      } else {
    64  {{ $typelist := .In }}
    65  {{range .In}}
    66  {{ $src := .Type -}}
    67  {{ $srcName := .Name -}}
    68  {{ range $typelist -}}
    69  {{ $dest := .Type -}}
    70  {{ $destName := .Name -}}
    71          Transpose{{$srcName}}{{$destName}} = transpose{{ $srcName }}{{ $destName }}
    72  {{end}}
    73  {{end}}    
    74      }
    75  }