github.com/apache/arrow/go/v14@v14.0.2/internal/utils/transpose_ints_simd.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  	"unsafe"
    24  )
    25  
    26  {{ $arch := .D.arch}}
    27  {{ $typelist := .In}}
    28  {{range .In}}
    29  {{ $src := .Type }}
    30  {{ $srcName := .Name }}
    31  {{ range $typelist}}
    32  {{ $dest := .Type }}
    33  {{ $destName := .Name }}
    34  
    35  //go:noescape
    36  func _transpose_{{printf "%s_%s_%s" $src $dest $arch}}(src, dest unsafe.Pointer, length int, transposeMap unsafe.Pointer)
    37  
    38  func transpose{{ $srcName }}{{ $destName }}{{ $arch }}(src []{{$src}}, dest []{{$dest}}, transposeMap []int32) {
    39  	_transpose_{{printf "%s_%s_%s" $src $dest $arch}}(unsafe.Pointer(&src[0]), unsafe.Pointer(&dest[0]), len(dest), unsafe.Pointer(&transposeMap[0]))
    40  }
    41  {{ end }}
    42  {{ end }}