Skip to content

Commit 184f361

Browse files
authored
remove sorting of bool values since torch 2.4 does not support it on cuda (flagos-ai#731)
1 parent 980657f commit 184f361

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

benchmark/test_special_perf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def sort_input_fn(shape, dtype, device):
128128
input_fn=sort_input_fn,
129129
op_name="sort",
130130
torch_op=torch.sort,
131-
dtypes=BOOL_DTYPES + INT_DTYPES + FLOAT_DTYPES,
131+
dtypes=INT_DTYPES + FLOAT_DTYPES,
132132
)
133133
bench.run()
134134

0 commit comments

Comments
 (0)