Skip to content

Commit 3c88f9e

Browse files
committed
Supress warnings for loopy flop calc
1 parent 5dec18a commit 3c88f9e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pyop2/base.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3608,7 +3608,9 @@ def num_flops(self):
36083608
return v.visit(self.code)
36093609
elif isinstance(self.code, loopy.LoopKernel):
36103610
op_map = loopy.get_op_map(
3611-
self.code.copy(options=loopy.Options(ignore_boostable_into=True)),
3611+
self.code.copy(options=loopy.Options(ignore_boostable_into=True),
3612+
silenced_warnings=['insn_count_subgroups_upper_bound',
3613+
'get_x_map_guessing_subgroup_size']),
36123614
subgroup_size='guess')
36133615
return op_map.filter_by(name=['add', 'sub', 'mul', 'div'], dtype=[ScalarType]).eval_and_sum({})
36143616
else:

0 commit comments

Comments
 (0)