bifrost: Honor src swizzle in special math ops

[mesa.git] / src / panfrost / bifrost / bifrost_nir_algebraic.py

diff --git a/src/panfrost/bifrost/bifrost_nir_algebraic.py b/src/panfrost/bifrost/bifrost_nir_algebraic.py
index 5e2c65c68ac59bf68eb3a24102fd2214586cf91a..6f3f6eee1a901fe5a214322ac2b9549f973267ed 100644 (file)
--- a/src/panfrost/bifrost/bifrost_nir_algebraic.py
+++ b/src/panfrost/bifrost/bifrost_nir_algebraic.py
@@ -86,6 +86,10 @@ SPECIAL = ['fexp2', 'flog2', 'fsin', 'fcos']
 for op in SPECIAL:
         converts += [((op + '@16', a), ('f2f16', (op, ('f2f32', a))))]
 
+converts += [(('f2b32', a), ('fne32', a, 0.0)),
+             (('i2b32', a), ('ine32', a, 0)),
+             (('b2i32', a), ('iand', 'a@32', 1))]
+
 def main():
     parser = argparse.ArgumentParser()
     parser.add_argument('-p', '--import-path', required=True)