diff --git a/src/tutorials/average_pool2d/average_pool2d_nki_kernels.py b/src/tutorials/average_pool2d/average_pool2d_nki_kernels.py index 5dfe606..c81a4a5 100644 --- a/src/tutorials/average_pool2d/average_pool2d_nki_kernels.py +++ b/src/tutorials/average_pool2d/average_pool2d_nki_kernels.py @@ -83,7 +83,7 @@ def np_average_pool_2D(in_tensor, pool_size): print(in_tensor, out_nki, out_np) - match = (out_nki == out_nki).all() + match = (out_nki == out_np).all() if match: print("NKI and NumPy match")