Skip to content

Commit

Permalink
fix tests and notebook
Browse files Browse the repository at this point in the history
  • Loading branch information
havogt committed Dec 6, 2024
1 parent 404cba1 commit 33f8b6b
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 7 deletions.
2 changes: 1 addition & 1 deletion docs/user/next/advanced/HackTheToolchain.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ from gt4py import eve

```python
cached_lowering_toolchain = gtx.backend.DEFAULT_TRANSFORMS.replace(
past_to_itir=gtx.ffront.past_to_itir.past_to_itir_factory(cached=False)
past_to_itir=gtx.ffront.past_to_itir.past_to_gtir_factory(cached=False)
)
```

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
)


def test_program_itir_regression(cartesian_case):
def test_program_gtir_regression(cartesian_case):
@gtx.field_operator(backend=None)
def testee_op(a: cases.IField) -> cases.IField:
return a
Expand All @@ -30,8 +30,8 @@ def testee_op(a: cases.IField) -> cases.IField:
def testee(a: cases.IField, out: cases.IField):
testee_op(a, out=out)

assert isinstance(testee.itir, itir.Program)
assert isinstance(testee.with_backend(cartesian_case.backend).itir, itir.Program)
assert isinstance(testee.gtir, itir.Program)
assert isinstance(testee.with_backend(cartesian_case.backend).gtir, itir.Program)


def test_frozen(cartesian_case):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -107,12 +107,12 @@ def test_verification(testee, run_gtfn_with_temporaries_and_symbolic_sizes, mesh


def test_temporary_symbols(testee, mesh_descriptor):
itir_with_tmp = apply_common_transforms(
testee.itir,
gtir_with_tmp = apply_common_transforms(
testee.gtir,
extract_temporaries=True,
offset_provider=mesh_descriptor.offset_provider,
)

params = ["num_vertices", "num_edges", "num_cells"]
for param in params:
assert any([param == str(p) for p in itir_with_tmp.params])
assert any([param == str(p) for p in gtir_with_tmp.params])

0 comments on commit 33f8b6b

Please sign in to comment.