
   0 | {--
   1 | Copyright (C) 2025  Joel Berkeley
   2 | 
   3 | This program is free software: you can redistribute it and/or modify
   4 | it under the terms of the GNU Affero General Public License as published
   5 | by the Free Software Foundation, either version 3 of the License, or
   6 | (at your option) any later version.
   7 | 
   8 | This program is distributed in the hope that it will be useful,
   9 | but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 | GNU Affero General Public License for more details.
  12 | 
  13 | You should have received a copy of the GNU Affero General Public License
  14 | along with this program.  If not, see <https://www.gnu.org/licenses/>.
  15 | --}
  16 | ||| For internal spidr use only.
  17 | module Spidr.Compiler.MLIR.IR.BuiltinAttributes
  18 | 
  19 | import Spidr.Compiler.LLVM.ADT.APFloat
  20 | import Spidr.Compiler.LLVM.ADT.APInt
  21 | import Spidr.Compiler.MLIR.IR.Attributes
  22 | import Spidr.Compiler.MLIR.IR.BuiltinTypes
  23 | import Spidr.Compiler.MLIR.IR.Types
  24 | import Spidr.Compiler.Array
  25 | import Spidr.Compiler.FFI
  26 | 
  27 | ffi : String -> String
  28 | ffi = libxla "c/mlir/IR/BuiltinAttributes.h"
  29 | 
  30 | public export
  31 | data DenseElementsAttr = MkDenseElementsAttr GCAnyPtr
  32 | 
  33 | export
  34 | %foreign (ffi "DenseElementsAttr_delete")
  35 | prim__deleteDenseElementsAttr : AnyPtr -> PrimIO ()
  36 | 
  37 | namespace DenseElementsAttr
  38 |   namespace ArrayRef
  39 |     namespace Attribute
  40 |       %foreign (ffi "DenseElementsAttr_get_ArrayRef_Attribute")
  41 |       prim__denseElementsAttrGet : GCAnyPtr -> AnyPtr -> Bits64 -> PrimIO AnyPtr
  42 | 
  43 |       export
  44 |       get : HasIO io => RankedTensorType -> List Attribute -> io DenseElementsAttr
  45 |       get (MkRankedTensorType rtt) _ = do
  46 |         attr <- primIO $ prim__denseElementsAttrGet rtt prim__getNullAnyPtr 0
  47 |         attr <- onCollectAny' attr (primIO . prim__deleteDenseElementsAttr)
  48 |         pure (MkDenseElementsAttr attr)
  49 | 
  50 |     getAux :
  51 |       (GCAnyPtr -> GCAnyPtr -> Bits64 -> PrimIO AnyPtr) ->
  52 |       HasIO io => RankedTensorType -> Array a -> io DenseElementsAttr
  53 |     getAux f (MkRankedTensorType rtt) (MkArray values len) = do
  54 |       attr <- primIO $ f rtt values len
  55 |       attr <- onCollectAny' attr (primIO . prim__deleteDenseElementsAttr)
  56 |       pure (MkDenseElementsAttr attr)
  57 | 
  58 |     namespace Int8
  59 |       %foreign (ffi "DenseElementsAttr_get_ArrayRef_int8_t")
  60 |       prim__denseElementsAttrGet : GCAnyPtr -> GCAnyPtr -> Bits64 -> PrimIO AnyPtr
  61 | 
  62 |       export
  63 |       get : HasIO io => RankedTensorType -> Array Int8 -> io DenseElementsAttr
  64 |       get = getAux prim__denseElementsAttrGet
  65 | 
  66 |     namespace Int16
  67 |       %foreign (ffi "DenseElementsAttr_get_ArrayRef_int16_t")
  68 |       prim__denseElementsAttrGet : GCAnyPtr -> GCAnyPtr -> Bits64 -> PrimIO AnyPtr
  69 | 
  70 |       export
  71 |       get : HasIO io => RankedTensorType -> Array Int16 -> io DenseElementsAttr
  72 |       get = getAux prim__denseElementsAttrGet
  73 | 
  74 |     namespace Int32
  75 |       %foreign (ffi "DenseElementsAttr_get_ArrayRef_int32_t")
  76 |       prim__denseElementsAttrGet : GCAnyPtr -> GCAnyPtr -> Bits64 -> PrimIO AnyPtr
  77 | 
  78 |       export
  79 |       get : HasIO io => RankedTensorType -> Array Int32 -> io DenseElementsAttr
  80 |       get = getAux prim__denseElementsAttrGet
  81 | 
  82 |     namespace Int64
  83 |       %foreign (ffi "DenseElementsAttr_get_ArrayRef_int64_t")
  84 |       prim__denseElementsAttrGet : GCAnyPtr -> GCAnyPtr -> Bits64 -> PrimIO AnyPtr
  85 | 
  86 |       export
  87 |       get : HasIO io => RankedTensorType -> Array Int64 -> io DenseElementsAttr
  88 |       get = getAux prim__denseElementsAttrGet
  89 | 
  90 |     namespace UInt8
  91 |       %foreign (ffi "DenseElementsAttr_get_ArrayRef_uint8_t")
  92 |       prim__denseElementsAttrGet : GCAnyPtr -> GCAnyPtr -> Bits64 -> PrimIO AnyPtr
  93 | 
  94 |       export
  95 |       get : HasIO io => RankedTensorType -> Array Bits8 -> io DenseElementsAttr
  96 |       get = getAux prim__denseElementsAttrGet
  97 | 
  98 |     namespace UInt16
  99 |       %foreign (ffi "DenseElementsAttr_get_ArrayRef_uint16_t")
 100 |       prim__denseElementsAttrGet : GCAnyPtr -> GCAnyPtr -> Bits64 -> PrimIO AnyPtr
 101 | 
 102 |       export
 103 |       get : HasIO io => RankedTensorType -> Array Bits16 -> io DenseElementsAttr
 104 |       get = getAux prim__denseElementsAttrGet
 105 | 
 106 |     namespace UInt32
 107 |       %foreign (ffi "DenseElementsAttr_get_ArrayRef_uint32_t")
 108 |       prim__denseElementsAttrGet : GCAnyPtr -> GCAnyPtr -> Bits64 -> PrimIO AnyPtr
 109 | 
 110 |       export
 111 |       get : HasIO io => RankedTensorType -> Array Bits32 -> io DenseElementsAttr
 112 |       get = getAux prim__denseElementsAttrGet
 113 | 
 114 |     namespace UInt64
 115 |       %foreign (ffi "DenseElementsAttr_get_ArrayRef_uint64_t")
 116 |       prim__denseElementsAttrGet : GCAnyPtr -> GCAnyPtr -> Bits64 -> PrimIO AnyPtr
 117 | 
 118 |       export
 119 |       get : HasIO io => RankedTensorType -> Array Bits64 -> io DenseElementsAttr
 120 |       get = getAux prim__denseElementsAttrGet
 121 | 
 122 |     namespace Double
 123 |       %foreign (ffi "DenseElementsAttr_get_ArrayRef_double")
 124 |       prim__denseElementsAttrGet : GCAnyPtr -> GCAnyPtr -> Bits64 -> PrimIO AnyPtr
 125 | 
 126 |       export
 127 |       get : HasIO io => RankedTensorType -> Array Double -> io DenseElementsAttr
 128 |       get = getAux prim__denseElementsAttrGet
 129 | 
 130 |     namespace Bool
 131 |       %foreign (ffi "DenseElementsAttr_get_ArrayRef_bool")
 132 |       prim__denseElementsAttrGet : GCAnyPtr -> GCAnyPtr -> Bits64 -> PrimIO AnyPtr
 133 | 
 134 |       export
 135 |       get : HasIO io => RankedTensorType -> Array Bool -> io DenseElementsAttr
 136 |       get = getAux prim__denseElementsAttrGet
 137 | 
 138 |   namespace APInt
 139 |     %foreign (ffi "DenseElementsAttr_get_APInt")
 140 |     prim__denseElementsAttrGet : GCAnyPtr -> GCAnyPtr -> PrimIO AnyPtr
 141 | 
 142 |     export
 143 |     get : HasIO io => RankedTensorType -> APInt -> io DenseElementsAttr
 144 |     get (MkRankedTensorType rtt) (MkAPInt value) = do
 145 |       attr <- primIO $ prim__denseElementsAttrGet rtt value
 146 |       attr <- onCollectAny' attr (primIO . prim__deleteDenseElementsAttr)
 147 |       pure (MkDenseElementsAttr attr)
 148 | 
 149 |   namespace APFloat
 150 |     %foreign (ffi "DenseElementsAttr_get_APFloat")
 151 |     prim__denseElementsAttrGet : GCAnyPtr -> GCAnyPtr -> PrimIO AnyPtr
 152 | 
 153 |     export
 154 |     get : HasIO io => RankedTensorType -> APFloat -> io DenseElementsAttr
 155 |     get (MkRankedTensorType rtt) (MkAPFloat value) = do
 156 |       attr <- primIO $ prim__denseElementsAttrGet rtt value
 157 |       attr <- onCollectAny' attr (primIO . prim__deleteDenseElementsAttr)
 158 |       pure (MkDenseElementsAttr attr)
 159 |