Skip to content

Commit def0d6a

Browse files
4vtomatkito-cheng
authored andcommitted
Fix widen type for bfloat in TypeHelper
Widened type for bfloat should be float
1 parent b467d79 commit def0d6a

File tree

1 file changed

+8
-3
lines changed
  • rvv-intrinsic-generator/rvv_intrinsic_gen

1 file changed

+8
-3
lines changed

rvv-intrinsic-generator/rvv_intrinsic_gen/utils.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,11 @@ class TypeHelper:
5959

6060
def __init__(self, **kargs):
6161
self.args = kargs
62+
if "TYPE" not in self.args:
63+
return
64+
self.args["WTYPE"] = self.args["TYPE"]
65+
if self.args["TYPE"] == "bfloat":
66+
self.args["WTYPE"] = "float"
6267

6368
@property
6469
def get_float_lmul(self):
@@ -154,15 +159,15 @@ def sis(self):
154159

155160
@property
156161
def wv(self):
157-
return "v{TYPE}{WSEW}m{WLMUL}_t".format_map(self.args)
162+
return "v{WTYPE}{WSEW}m{WLMUL}_t".format_map(self.args)
158163

159164
@property
160165
def wvm1(self):
161-
return "v{TYPE}{WSEW}m1_t".format_map(self.args)
166+
return "v{WTYPE}{WSEW}m1_t".format_map(self.args)
162167

163168
@property
164169
def ws(self):
165-
return "{TYPE}{WSEW}_t".format_map(self.args)
170+
return "{WTYPE}{WSEW}_t".format_map(self.args)
166171

167172
@property
168173
def qv(self):

0 commit comments

Comments
 (0)