add defintrinsic

.formatter.exs

@@ -3,7 +3,8 @@ locals_without_parens = [
   op: 1,
   value: 1,
   call: 1,
-  const: 1
+  const: 1,
+  defintrinsic: 1
 ]
 
 [

lib/charms/defm/expander.ex

@@ -477,7 +477,7 @@ defmodule Charms.Defm.Expander do
             Code.ensure_loaded(module)
 
             cond do
-              function_exported?(module, :handle_intrinsic, 3) ->
+              function_exported?(module, :__intrinsics__, 0) and fun in module.__intrinsics__() ->
                 {args, state, env} = expand(args, state, env)
 
                 {module.handle_intrinsic(fun, args, ctx: state.mlir.ctx, block: state.mlir.blk),

lib/charms/env.ex

@@ -8,4 +8,6 @@ defmodule Charms.Env do
   def handle_intrinsic(:t, [], opts) do
     Beaver.ENIF.Type.env(opts)
   end
+
+  defintrinsic [:t]
 end

lib/charms/intrinsic.ex

@@ -6,11 +6,38 @@ defmodule Charms.Intrinsic do
   @type opt :: {:ctx, MLIR.Context.t()} | {:block, MLIR.Block.t()}
   @type opts :: [opt | {atom(), term()}]
   @callback handle_intrinsic(atom(), [term()], opts()) :: term()
+  Module.register_attribute(__MODULE__, :defintrinsic, accumulate: true)
+
+  @doc false
+  def collect_intrinsics(nil) do
+    raise ArgumentError, "no intrinsic functions defined"
+  end
+
+  def collect_intrinsics(attr_list) when length(attr_list) > 0 do
+    attr_list |> Enum.reverse() |> List.flatten() |> Enum.uniq()
+  end
 
   defmacro __using__(_) do
     quote do
       @behaviour Charms.Intrinsic
       use Beaver
+      @before_compile Charms.Intrinsic
+      import Charms.Intrinsic, only: :macros
+    end
+  end
+
+  defmacro defintrinsic(intrinsic_list) do
+    quote do
+      @defintrinsic unquote(intrinsic_list)
+    end
+  end
+
+  defmacro __before_compile__(_env) do
+    quote do
+      @defintrinsic_list @defintrinsic |> Charms.Intrinsic.collect_intrinsics()
+      def __intrinsics__() do
+        @defintrinsic_list
+      end
     end
   end
 end

lib/charms/pointer.ex

@@ -55,4 +55,6 @@ defmodule Charms.Pointer do
   def handle_intrinsic(:t, [], opts) do
     Beaver.Deferred.from_opts(opts, ~t{!llvm.ptr})
   end
+
+  defintrinsic [:t, :allocate, :load, :store, :element_ptr]
 end

lib/charms/prelude.ex

@@ -110,4 +110,6 @@ defmodule Charms.Prelude do
   def handle_intrinsic(_name, _args, _opts) do
     :not_handled
   end
+
+  defintrinsic [:result_at, :!=, :-, :+, :<, :>, :<=, :>=, :==, :&&, :*]
 end

lib/charms/simd.ex

@@ -26,4 +26,6 @@ defmodule Charms.SIMD do
   def handle_intrinsic(:t, [type, width], _opts) do
     Type.vector([width], type)
   end
+
+  defintrinsic [:new, :t]
 end

lib/charms/term.ex

@@ -8,4 +8,6 @@ defmodule Charms.Term do
   def handle_intrinsic(:t, [], opts) do
     Beaver.ENIF.Type.term(opts)
   end
+
+  defintrinsic [:t]
 end