fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2025-12-28 10:20:09 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

nak: Add an SSARef::from_iter() helper

Browse source 
Reviewed-by: Faith Ekstrand <faith.ekstrand@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/34794>
This commit is contained in:
Mel Henning 2025-04-29 21:51:23 -04:00 • committed by Marge Bot
parent 4f07092bdb
commit 121b2b889b
1 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/nouveau/compiler/nak/ir.rs
View file

@ -14,6 +14,7 @@ use compiler::as_slice::*;
use compiler::cfg::CFG;
use compiler::smallvec::SmallVec;
use nak_ir_proc::*;
use std::array;
use std::cmp::{max, min};
use std::fmt;
use std::fmt::Write;
@ -447,6 +448,14 @@ impl SSARef {
r
}
fn from_iter(mut it: impl ExactSizeIterator<Item = SSAValue>) -> Self {
let len = it.len();
assert!(len > 0 && len <= 4);
let v: [SSAValue; 4] =
array::from_fn(|_| it.next().unwrap_or(SSAValue::NONE));
Self::new(&v[..len])
}
/// Returns the number of components in this SSA reference
pub fn comps(&self) -> u8 {
if self.v[3].packed >= u32::MAX - 2 {
@ -592,12 +601,7 @@ impl SSAValueAllocator {
}
pub fn alloc_vec(&mut self, file: RegFile, comps: u8) -> SSARef {
assert!(comps >= 1 && comps <= 4);
let mut vec = [SSAValue::NONE; 4];
for c in 0..comps {
vec[usize::from(c)] = self.alloc(file);
}
vec[0..usize::from(comps)].try_into().unwrap()
SSARef::from_iter((0..comps).map(|_| self.alloc(file)))
}
}