loki.transformations.tests.test_loop_blocking

Functions

test_1d_blocking(tmp_path, frontend, ...)

Apply loop blocking of simple loops into two loops

test_1d_blocking_multi_intent(tmp_path, ...)

Apply loop blocking of simple loops into two loops

test_1d_splitting(tmp_path, frontend, ...)

Apply loop blocking of simple loops into two loops

test_1d_splitting_multi_var(tmp_path, ...)

Apply loop blocking of simple loops into two loops

test_2d_blocking(tmp_path, frontend, ...)

test_2d_splitting(tmp_path, frontend, ...)

test_3d_blocking(tmp_path, frontend, ...)

test_3d_splitting(tmp_path, frontend, ...)

test_1d_splitting(tmp_path, frontend, block_size, n)

Apply loop blocking of simple loops into two loops

test_1d_splitting_multi_var(tmp_path, frontend, block_size, n)

Apply loop blocking of simple loops into two loops

test_2d_splitting(tmp_path, frontend, block_size, n)
test_3d_splitting(tmp_path, frontend, block_size, n)
test_1d_blocking(tmp_path, frontend, block_size, n)

Apply loop blocking of simple loops into two loops

test_1d_blocking_multi_intent(tmp_path, frontend, block_size, n)

Apply loop blocking of simple loops into two loops

test_2d_blocking(tmp_path, frontend, block_size, n)
test_3d_blocking(tmp_path, frontend, block_size, n)