Skip to content

Commit fdbbe6e

Browse files
committed
Update rope.py to keep up with NineToothed 0.16.0 updates
1 parent 19dbae5 commit fdbbe6e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

rope.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def application(tensor, sin_table, cos_table):
3838
tensor[1] = tensor_0 * sin_table + tensor_1 * cos_table
3939

4040

41-
tensors = tuple(Tensor(4, constexpr_shape=True) for _ in range(3))
41+
tensors = tuple(Tensor(4, shape_options={"constexpr": True}) for _ in range(3))
4242
rope_kernel = ninetoothed.make(arrangement, application, tensors)
4343

4444

0 commit comments

Comments
 (0)