diff --git a/src/nouveau/compiler/nak/ir.rs b/src/nouveau/compiler/nak/ir.rs index 748eb618178..4b82652c874 100644 --- a/src/nouveau/compiler/nak/ir.rs +++ b/src/nouveau/compiler/nak/ir.rs @@ -8769,10 +8769,20 @@ pub trait ShaderModel { } #[allow(dead_code)] - fn is_blackwell(&self) -> bool { + fn is_blackwell_a(&self) -> bool { self.sm() >= 100 && self.sm() < 110 } + #[allow(dead_code)] + fn is_blackwell_b(&self) -> bool { + self.sm() >= 120 && self.sm() < 130 + } + + #[allow(dead_code)] + fn is_blackwell(&self) -> bool { + self.is_blackwell_a() || self.is_blackwell_b() + } + fn num_regs(&self, file: RegFile) -> u32; fn hw_reserved_gprs(&self) -> u32; fn crs_size(&self, max_crs_depth: u32) -> u32;