feat: add amo instruction layout system for Zaamo & Zabha extension & some Zacas extension

Rakefile

@@ -5,6 +5,9 @@ require "sorbet-runtime"
 T.bind(self, T.all(Rake::DSL, Object))
 extend T::Sig
 
+require 'pathname'
+require 'erb'
+
 Encoding.default_external = "UTF-8"
 
 $jobs = ENV["JOBS"].nil? ? 1 : ENV["JOBS"].to_i
@@ -155,8 +158,38 @@ end
 
 
 namespace :test do
+  # Build list of all AMO instruction files that need to be generated
+  amo_instruction_files = []
+
+  # Zaamo and Zabha AMO variants
+  %w[amoadd amoand amomax amomaxu amomin amominu amoor amoswap amoxor].each do |op|
+    ["b", "h", "w", "d"].each do |size|
+      extension_dir = %w[b h].include?(size) ? "Zabha" : "Zaamo"
+      [
+        { suffix: "", aq: false, rl: false },
+        { suffix: ".aq", aq: true, rl: false },
+        { suffix: ".rl", aq: false, rl: true },
+        { suffix: ".aqrl", aq: true, rl: true }
+      ].each do |variant|
+        amo_instruction_files << "#{$resolver.std_path}/inst/#{extension_dir}/#{op}.#{size}#{variant[:suffix]}.yaml"
+      end
+    end
+  end
+
+  # AMOCAS variants for Zabha (b, h sizes)
+  ["b", "h"].each do |size|
+    [
+      { suffix: "", aq: false, rl: false },
+      { suffix: ".aq", aq: true, rl: false },
+      { suffix: ".rl", aq: false, rl: true },
+      { suffix: ".aqrl", aq: true, rl: true }
+    ].each do |variant|
+      amo_instruction_files << "#{$resolver.std_path}/inst/Zabha/amocas.#{size}#{variant[:suffix]}.yaml"
+    end
+  end
+
   desc "Check that instruction encodings in the DB are consistent and do not conflict"
-  task :inst_encodings do
+  task :inst_encodings => amo_instruction_files do
     print "Checking for conflicts in instruction encodings.."
 
     cfg_arch = $resolver.cfg_arch_for("_")
@@ -356,9 +389,88 @@ file "#{$resolver.std_path}/csr/Zicntr/mcountinhibit.yaml" => [
   File.write(t.name, insert_warning(erb.result(binding), t.prerequisites.first))
 end
 
+# AMO instruction generation from layouts
+%w[amoadd amoand amomax amomaxu amomin amominu amoor amoswap amoxor].each do |op|
+  ["b", "h", "w", "d"].each do |size|
+    # Determine target extension directory based on size
+    extension_dir = %w[b h].include?(size) ? "Zabha" : "Zaamo"
+
+    # Define all acquire/release combinations
+    aq_rl_variants = [
+      { suffix: "", aq: false, rl: false },           # base instruction
+      { suffix: ".aq", aq: true, rl: false },         # acquire only
+      { suffix: ".rl", aq: false, rl: true },         # release only
+      { suffix: ".aqrl", aq: true, rl: true }         # both acquire and release
+    ]
+
+    aq_rl_variants.each do |variant|
+      file "#{$resolver.std_path}/inst/#{extension_dir}/#{op}.#{size}#{variant[:suffix]}.yaml" => [
+        "#{$resolver.std_path}/inst/Zaamo/#{op}.SIZE.AQRL.layout",
+        __FILE__
+      ] do |t|
+        aq = variant[:aq]
+        rl = variant[:rl]
+        erb = ERB.new(File.read($resolver.std_path / "inst/Zaamo/#{op}.SIZE.AQRL.layout"), trim_mode: "-")
+        erb.filename = "#{$resolver.std_path}/inst/Zaamo/#{op}.SIZE.AQRL.layout"
+        File.write(t.name, insert_warning(erb.result(binding), t.prerequisites.first))
+      end
+    end
+  end
+end
+
+# AMOCAS instruction generation from Zabha layout (supports both Zabha and Zacas)
+# Zabha variants (b, h) -> generated in Zabha directory
+["b", "h"].each do |size|
+  # Define all acquire/release combinations
+  aq_rl_variants = [
+    { suffix: "", aq: false, rl: false },           # base instruction
+    { suffix: ".aq", aq: true, rl: false },         # acquire only
+    { suffix: ".rl", aq: false, rl: true },         # release only
+    { suffix: ".aqrl", aq: true, rl: true }         # both acquire and release
+  ]
+
+  aq_rl_variants.each do |variant|
+    file "#{$resolver.std_path}/inst/Zabha/amocas.#{size}#{variant[:suffix]}.yaml" => [
+      "#{$resolver.std_path}/inst/Zabha/amocas.SIZE.AQRL.layout",
+      __FILE__
+    ] do |t|
+      aq = variant[:aq]
+      rl = variant[:rl]
+      erb = ERB.new(File.read($resolver.std_path / "inst/Zabha/amocas.SIZE.AQRL.layout"), trim_mode: "-")
+      erb.filename = "#{$resolver.std_path}/inst/Zabha/amocas.SIZE.AQRL.layout"
+      File.write(t.name, insert_warning(erb.result(binding), t.prerequisites.first))
+    end
+  end
+end
+
+# Zacas variants (w, d, q) -> generated in Zacas directory using the same Zabha layout
+["w", "d", "q"].each do |size|
+  # Define all acquire/release combinations
+  aq_rl_variants = [
+    { suffix: "", aq: false, rl: false },           # base instruction
+    { suffix: ".aq", aq: true, rl: false },         # acquire only
+    { suffix: ".rl", aq: false, rl: true },         # release only
+    { suffix: ".aqrl", aq: true, rl: true }         # both acquire and release
+  ]
+
+  aq_rl_variants.each do |variant|
+    file "#{$resolver.std_path}/inst/Zacas/amocas.#{size}#{variant[:suffix]}.yaml" => [
+      "#{$resolver.std_path}/inst/Zabha/amocas.SIZE.AQRL.layout",
+      __FILE__
+    ] do |t|
+      aq = variant[:aq]
+      rl = variant[:rl]
+      erb = ERB.new(File.read($resolver.std_path / "inst/Zabha/amocas.SIZE.AQRL.layout"), trim_mode: "-")
+      erb.filename = "#{$resolver.std_path}/inst/Zabha/amocas.SIZE.AQRL.layout"
+      File.write(t.name, insert_warning(erb.result(binding), t.prerequisites.first))
+    end
+  end
+end
+
 namespace :gen do
   desc "Generate architecture files from layouts"
   task :arch do
+    # Generate CSR files
     (3..31).each do |hpm_num|
       Rake::Task["#{$resolver.std_path}/csr/Zihpm/mhpmcounter#{hpm_num}.yaml"].invoke
       Rake::Task["#{$resolver.std_path}/csr/Zihpm/mhpmcounter#{hpm_num}h.yaml"].invoke