diff --git a/mobile/tools/python/fluidtools/run.py b/mobile/tools/python/fluidtools/run.py index 4773222536a53f28f433b2538d617233d05ab685..6f82e426bd1ab1e376783c0d1015e625d7d47068 100644 --- a/mobile/tools/python/fluidtools/run.py +++ b/mobile/tools/python/fluidtools/run.py @@ -559,7 +559,7 @@ def check_mobile_results(args, fuse, mem_opt): for i in range(len(values1)): v1 = values1[i] v2 = values2[len(shape) + i] - if abs(v1 - v2) > diff_threshold: + if ((not math.isnan(v1)) and math.isnan(v2)) or abs(v1 - v2) > diff_threshold: error_index = index break checked_names.append(op_output_var_name)