github/wgpu
1
1
Fork 0
mirror of https://github.com/gfx-rs/wgpu.git synced 2026-04-22 03:02:01 -04:00
Code Issues Packages Projects Releases Wiki Activity

[spv-in,spv-out] Handle outer product

Browse Source
This commit is contained in:
Ashley
2021-05-04 18:50:37 +02:00
committed by Dzmitry Malyshau
parent 04c639c3f1
commit 3ffc361724
2 changed files with 31 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/back/spv/writer.rs
View File

@@ -1815,6 +1815,13 @@ impl Writer {
arg0_id,
arg1_id,
)),
Mf::Outer => MathOp::Custom(Instruction::binary(
spirv::Op::OuterProduct,
result_type_id,
id,
arg0_id,
arg1_id,
)),
Mf::Cross => MathOp::Ext(spirv::GLOp::Cross),
Mf::Distance => MathOp::Ext(spirv::GLOp::Distance),
Mf::Length => MathOp::Ext(spirv::GLOp::Length),
@@ -1848,7 +1855,7 @@ impl Writer {
arg0_id,
)),
Mf::Determinant => MathOp::Ext(spirv::GLOp::Determinant),
Mf::Outer | Mf::ReverseBits | Mf::CountOneBits => {
Mf::ReverseBits | Mf::CountOneBits => {
log::error!("unimplemented math function {:?}", fun);
return Err(Error::FeatureNotImplemented("math function"));
}

23
src/front/spv/mod.rs
View File

@@ -1426,6 +1426,29 @@ impl<I: Iterator<Item = u32>> Parser<I> {
},
);
}
Op::OuterProduct => {
inst.expect(5)?;
let result_type_id = self.next()?;
let result_id = self.next()?;
let left_id = self.next()?;
let right_id = self.next()?;
let left_lexp = self.lookup_expression.lookup(left_id)?;
let right_lexp = self.lookup_expression.lookup(right_id)?;
let expr = crate::Expression::Math {
fun: crate::MathFunction::Outer,
arg: left_lexp.handle,
arg1: Some(right_lexp.handle),
arg2: None,
};
self.lookup_expression.insert(
result_id,
LookupExpression {
handle: expressions.append(expr),
type_id: result_type_id,
},
);
}
// Bitwise instructions
Op::Not => {
inst.expect(4)?;