Comparison Functions

Module: polygraphy.tools.args

class CompareFuncSimpleArgs[source]

Bases: BaseArgs

Comparison Function: simple: the CompareFunc.simple comparison function.

Depends on:

ComparatorCompareArgs

parse_impl(args)[source]

Parses command-line arguments and populates the following attributes:

no_shape_check

Whether to skip shape checks.

rtol

Per-tensor relative tolerance.

atol

Per-tensor absolute tolerance.

check_error_stat

The error metric to check.

infinities_compare_equal

Whether to allow +-inf to compare as equal.

save_heatmaps

Directory in which to save heatmaps of error.

show_heatmaps

Whether to display heatmaps of error.

save_error_metrics_plot

Path to store generated error plots.

show_error_metrics_plot

Whether to display the error metrics plots.

error_quantile

Per-tensor quantile of error to compute.

class CompareFuncIndicesArgs[source]

Bases: BaseArgs

Comparison Function: indices: the CompareFunc.indices comparison function.

Depends on:

ComparatorCompareArgs

parse_impl(args)[source]

Parses command-line arguments and populates the following attributes:

index_tolerance

Per-tensor index tolerance.