From e95ab2dc50cc13d2c52e1b217e1fb3bf2880741b Mon Sep 17 00:00:00 2001 From: Kishan-Ved Date: Fri, 5 Jan 2024 13:45:38 +0530 Subject: [PATCH] fixed memory segfault --- integration_tests/elemental_01.py | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/integration_tests/elemental_01.py b/integration_tests/elemental_01.py index 668fa59a456..0a2787c7ede 100644 --- a/integration_tests/elemental_01.py +++ b/integration_tests/elemental_01.py @@ -9,17 +9,17 @@ def verify1d(array: f32[:], result: f32[:], size: i32): for i in range(size): assert abs(sin(sin(array[i])) - result[i]) <= eps -def verifynd(array: f32[:, :, :], result: f32[:, :, :], size1: i32, size2: i32, size3: i32): +def verifynd(array: f64[:, :, :], result: f64[:, :, :], size1: i32, size2: i32, size3: i32): i: i32 j: i32 k: i32 - eps: f32 - eps = f32(1e-6) + eps: f64 + eps = 1e-12 for i in range(size1): for j in range(size2): for k in range(size3): - assert abs(sin(array[i, j, k])**f32(2) - result[i, j, k]) <= eps + assert abs(sin(array[i, j, k])**2.0 - result[i, j, k]) <= eps def verify2d(array: f64[:, :], result: f64[:, :], size1: i32, size2: i32): i: i32 @@ -103,17 +103,17 @@ def elemental_sin(): verify1d(array1d, sin1d, 256) - arraynd: f32[256, 64, 16] = empty((256, 64, 16), dtype=float32) - sinnd: f32[256, 64, 16] = empty((256, 64, 16), dtype=float32) + arraynd: f64[200, 64, 16] = empty((200, 64, 16), dtype=float64) + sinnd: f64[200, 64, 16] = empty((200, 64, 16), dtype=float64) - for i in range(256): + for i in range(200): for j in range(64): for k in range(16): - arraynd[i, j, k] = f32(i + j + k) + arraynd[i, j, k] = float(i + j + k) - sinnd = sin(arraynd)**f32(2) + sinnd = sin(arraynd)**2.0 - verifynd(arraynd, sinnd, 256, 64, 16) + verifynd(arraynd, sinnd, 200, 64, 16) def elemental_cos(): i: i32 @@ -162,4 +162,4 @@ def elemental_trig_identity(): elemental_cos() elemental_trig_identity() elemental_sum() -elemental_mul() +elemental_mul() \ No newline at end of file