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 Compiler.Stablehlo.Dialect.Version
18 |
19 | import Compiler.FFI
20 |
21 | ffi : String -> String
22 | ffi = libxla "c/stablehlo/dialect/Version.h"
23 |
24 | public export
25 | data Version = MkVersion GCAnyPtr
26 |
27 | %foreign (ffi "Version_delete")
28 | prim__delete : AnyPtr -> PrimIO ()
29 |
30 | %foreign (ffi "Version_getCurrentVersion")
31 | prim__versionGetCurrentVersion : PrimIO AnyPtr
32 |
33 | export
34 | getCurrentVersion : HasIO io => io Version
35 | getCurrentVersion = do
36 |   version <- primIO $ prim__versionGetCurrentVersion
37 |   version <- onCollectAny' version (primIO . prim__delete)
38 |   pure (MkVersion version)
39 |
40 | %foreign (ffi "Version_toString")
41 | prim__versionToString : GCAnyPtr -> PrimIO AnyPtr
42 |
43 | export
44 | toString : HasIO io => Version -> io CppString
45 | toString (MkVersion version) = do
46 |   str <- primIO $ prim__versionToString version
47 |   pure (MkCppString str)
48 |