enable net_size_check by default with different max sizes for c and cuda

examples/hello_world.bend

@@ -1,4 +1,4 @@
 def main():
   with IO:
-    * <- IO/print("Hello, world!")
+    * <- IO/print("Hello, world!\n")
     return wrap(0)

src/hvm/check_net_size.rs

@@ -1,17 +1,27 @@
 use super::tree_children;
-use crate::{diagnostics::Diagnostics, fun::Name};
+use crate::{diagnostics::Diagnostics, fun::Name, CompileOpts};
 use hvm::ast::{Book, Net, Tree};
 
-pub const MAX_NET_SIZE: usize = 64;
+pub const MAX_NET_SIZE_C: usize = 4095;
+pub const MAX_NET_SIZE_CUDA: usize = 64;
+
+pub fn check_net_sizes(
+  book: &Book,
+  diagnostics: &mut Diagnostics,
+  opts: &CompileOpts,
+) -> Result<(), Diagnostics> {
+  let net_size_bound = match opts.command.as_str() {
+    "run-cu" | "gen-cu" | "gen-hvm" => MAX_NET_SIZE_CUDA,
+    _ => MAX_NET_SIZE_C,
+  };
 
-pub fn check_net_sizes(book: &Book, diagnostics: &mut Diagnostics) -> Result<(), Diagnostics> {
   diagnostics.start_pass();
 
   for (name, net) in &book.defs {
     let nodes = count_nodes(net);
-    if nodes > MAX_NET_SIZE {
+    if nodes > net_size_bound {
       diagnostics.add_rule_error(
-        format!("Definition is too large for hvm (size={nodes}, max size={MAX_NET_SIZE}). Please break it into smaller pieces."),
+        format!("Definition is too large for hvm (size={nodes}, max size={net_size_bound}). Please break it into smaller pieces."),
         Name::new(name),
       );
     }

src/lib.rs

@@ -2,7 +2,7 @@ use crate::{
   fun::{book_to_hvm, net_to_term::net_to_term, term_to_net::Labels, Book, Ctx, Term},
   hvm::{
     add_recursive_priority::add_recursive_priority,
-    check_net_size::{check_net_sizes, MAX_NET_SIZE},
+    check_net_size::{check_net_sizes, MAX_NET_SIZE_CUDA},
     eta_reduce::eta_reduce_hvm_net,
     hvm_book_show_pretty,
     inline::inline_hvm_book,
@@ -73,7 +73,7 @@ pub fn compile_book(
   }
 
   if opts.check_net_size {
-    check_net_sizes(&hvm_book, &mut diagnostics)?;
+    check_net_sizes(&hvm_book, &mut diagnostics, &opts)?;
   }
 
   add_recursive_priority(&mut hvm_book);
@@ -143,7 +143,7 @@ pub fn desugar_book(
   ctx.check_unbound_vars()?;
 
   if opts.float_combinators {
-    ctx.book.float_combinators(MAX_NET_SIZE);
+    ctx.book.float_combinators(MAX_NET_SIZE_CUDA);
   }
   // sanity check
   ctx.check_unbound_refs()?;
@@ -331,6 +331,9 @@ impl OptLevel {
 
 #[derive(Clone, Debug)]
 pub struct CompileOpts {
+  /// The Bend command
+  pub command: String,
+
   /// Enables [hvm::eta_reduce].
   pub eta: bool,
 
@@ -361,6 +364,7 @@ impl CompileOpts {
   #[must_use]
   pub fn set_all(self) -> Self {
     Self {
+      command: self.command,
       eta: true,
       prune: true,
       float_combinators: true,
@@ -376,6 +380,7 @@ impl CompileOpts {
   #[must_use]
   pub fn set_no_all(self) -> Self {
     Self {
+      command: self.command,
       eta: false,
       prune: false,
       linearize_matches: OptLevel::Disabled,
@@ -406,13 +411,14 @@ impl Default for CompileOpts {
   /// Uses num-scott ADT encoding.
   fn default() -> Self {
     Self {
+      command: String::from("run"),
       eta: true,
       prune: false,
       linearize_matches: OptLevel::Enabled,
       float_combinators: true,
       merge: false,
       inline: false,
-      check_net_size: false,
+      check_net_size: true,
       adt_encoding: AdtEncoding::NumScott,
     }
   }

src/main.rs

@@ -190,10 +190,14 @@ pub enum OptArgs {
   AdtNumScott,
 }
 
-fn compile_opts_from_cli(args: &Vec<OptArgs>) -> CompileOpts {
+fn compile_opts_from_cli(args: &Vec<OptArgs>, cmd: Option<&str>) -> CompileOpts {
   use OptArgs::*;
   let mut opts = CompileOpts::default();
 
+  if let Some(cmd) = cmd {
+    opts.command = cmd.to_string();
+  }
+
   for arg in args {
     match arg {
       All => opts = opts.set_all(),
@@ -291,7 +295,7 @@ fn execute_cli_mode(mut cli: Cli) -> Result<(), Diagnostics> {
   match cli.mode {
     Mode::Check { comp_opts, warn_opts, path } => {
       let diagnostics_cfg = set_warning_cfg_from_cli(DiagnosticsConfig::default(), warn_opts);
-      let compile_opts = compile_opts_from_cli(&comp_opts);
+      let compile_opts = compile_opts_from_cli(&comp_opts, None);
 
       let mut book = load_book(&path, diagnostics_cfg)?;
       let diagnostics = check_book(&mut book, diagnostics_cfg, compile_opts)?;
@@ -300,7 +304,7 @@ fn execute_cli_mode(mut cli: Cli) -> Result<(), Diagnostics> {
 
     Mode::GenHvm(GenArgs { comp_opts, warn_opts, path, .. }) => {
       let diagnostics_cfg = set_warning_cfg_from_cli(DiagnosticsConfig::default(), warn_opts);
-      let opts = compile_opts_from_cli(&comp_opts);
+      let opts = compile_opts_from_cli(&comp_opts, Some("gen-hvm"));
 
       let mut book = load_book(&path, diagnostics_cfg)?;
       let compile_res = compile_book(&mut book, opts, diagnostics_cfg, None)?;
@@ -317,7 +321,7 @@ fn execute_cli_mode(mut cli: Cli) -> Result<(), Diagnostics> {
       let diagnostics_cfg =
         set_warning_cfg_from_cli(DiagnosticsConfig::new(Severity::Allow, arg_verbose), warn_opts);
 
-      let compile_opts = compile_opts_from_cli(&comp_opts);
+      let compile_opts = compile_opts_from_cli(&comp_opts, Some(run_cmd));
 
       compile_opts.check_for_strict();
 
@@ -342,7 +346,7 @@ fn execute_cli_mode(mut cli: Cli) -> Result<(), Diagnostics> {
     Mode::GenC(GenArgs { comp_opts, warn_opts, path })
     | Mode::GenCu(GenArgs { comp_opts, warn_opts, path }) => {
       let diagnostics_cfg = set_warning_cfg_from_cli(DiagnosticsConfig::default(), warn_opts);
-      let opts = compile_opts_from_cli(&comp_opts);
+      let opts = compile_opts_from_cli(&comp_opts, Some(gen_cmd));
 
       let mut book = load_book(&path, diagnostics_cfg)?;
       let compile_res = compile_book(&mut book, opts, diagnostics_cfg, None)?;
@@ -373,7 +377,7 @@ fn execute_cli_mode(mut cli: Cli) -> Result<(), Diagnostics> {
     Mode::Desugar { path, comp_opts, warn_opts, pretty } => {
       let diagnostics_cfg = set_warning_cfg_from_cli(DiagnosticsConfig::default(), warn_opts);
 
-      let opts = compile_opts_from_cli(&comp_opts);
+      let opts = compile_opts_from_cli(&comp_opts, None);
 
       let mut book = load_book(&path, diagnostics_cfg)?;
       let diagnostics = desugar_book(&mut book, opts, diagnostics_cfg, None)?;

tests/golden_tests.rs

@@ -517,7 +517,7 @@ fn compile_long() {
       ..Default::default()
     };
 
-    compile_book(&mut book, opts.clone(), diagnostics_cfg, None)?;
-    Ok("Compiled".to_string())
+    let res = compile_book(&mut book, opts.clone(), diagnostics_cfg, None)?;
+    Ok(format!("{}{}", res.diagnostics, hvm_book_show_pretty(&res.hvm_book)))
   })
 }

tests/snapshots/compile_long__huge_tree.bend.snap

@@ -2,4 +2,6 @@
 source: tests/golden_tests.rs
 input_file: tests/golden_tests/compile_long/huge_tree.bend
 ---
-Compiled
+Errors:
+In definition 'main':
+  Definition is too large for hvm (size=120002, max size=4095). Please break it into smaller pieces.