Skip to content

Commit 54a5ffe

Browse files
authored
fix vector (#138)
1 parent ca63bbb commit 54a5ffe

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mlir/extras/dialects/ext/vector.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def __setitem__(self, idx, val):
7979

8080

8181
def transfer_write(
82-
vector: Vector,
82+
val: Vector,
8383
dest,
8484
indices,
8585
*,
@@ -92,13 +92,13 @@ def transfer_write(
9292
if loc is None:
9393
loc = get_user_code_loc()
9494
if permutation_map is None:
95-
permutation_map = AffineMap.get_minor_identity(dest.type.rank, vector.type.rank)
95+
permutation_map = AffineMap.get_minor_identity(dest.type.rank, val.type.rank)
9696
for j, i in enumerate(indices):
9797
if isinstance(i, int):
9898
indices[j] = constant(i, index=True)
9999
return _transfer_write(
100100
result=None,
101-
vector=vector,
101+
value_to_store=val,
102102
# no clue why they chose this name...
103103
source=dest,
104104
indices=indices,

0 commit comments

Comments
 (0)