Skip to content
This repository has been archived by the owner on Nov 27, 2024. It is now read-only.

Commit

Permalink
Supress warnings for loopy flop calc
Browse files Browse the repository at this point in the history
  • Loading branch information
wence- committed May 29, 2019
1 parent 5dec18a commit 3c88f9e
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion pyop2/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -3608,7 +3608,9 @@ def num_flops(self):
return v.visit(self.code)
elif isinstance(self.code, loopy.LoopKernel):
op_map = loopy.get_op_map(
self.code.copy(options=loopy.Options(ignore_boostable_into=True)),
self.code.copy(options=loopy.Options(ignore_boostable_into=True),
silenced_warnings=['insn_count_subgroups_upper_bound',
'get_x_map_guessing_subgroup_size']),
subgroup_size='guess')
return op_map.filter_by(name=['add', 'sub', 'mul', 'div'], dtype=[ScalarType]).eval_and_sum({})
else:
Expand Down

0 comments on commit 3c88f9e

Please sign in to comment.