Skip to content

Commit 97bfcb5

Browse files
committed
add note about flop counting for subscript indices
1 parent fe658de commit 97bfcb5

1 file changed

Lines changed: 15 additions & 0 deletions

File tree

pytato/analysis/__init__.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1439,6 +1439,11 @@ def get_num_flops(
14391439
this function assumes a SIMT-like model of computation in which the per-entry
14401440
cost is the sum of the costs of the two branches.
14411441
1442+
.. note::
1443+
1444+
Calculations for array subscripts are currently assumed to be integer-typed.
1445+
Any floating point operations contained within will be ignored.
1446+
14421447
.. note::
14431448
14441449
Does not support functions. Function calls must be inlined before calling.
@@ -1476,6 +1481,11 @@ def get_materialized_node_flop_counts(
14761481
this function assumes a SIMT-like model of computation in which the per-entry
14771482
cost is the sum of the costs of the two branches.
14781483
1484+
.. note::
1485+
1486+
Calculations for array subscripts are currently assumed to be integer-typed.
1487+
Any floating point operations contained within will be ignored.
1488+
14791489
.. note::
14801490
14811491
Does not support functions. Function calls must be inlined before calling.
@@ -1519,6 +1529,11 @@ def get_unmaterialized_node_flop_counts(
15191529
this function assumes a SIMT-like model of computation in which the per-entry
15201530
cost is the sum of the costs of the two branches.
15211531
1532+
.. note::
1533+
1534+
Calculations for array subscripts are currently assumed to be integer-typed.
1535+
Any floating point operations contained within will be ignored.
1536+
15221537
.. note::
15231538
15241539
Does not support functions. Function calls must be inlined before calling.

0 commit comments

Comments
 (0)