Skip to content

Commit

Permalink
requires_gpu
Browse files Browse the repository at this point in the history
  • Loading branch information
havogt committed Jan 25, 2024
1 parent db5ecfa commit ca7aa6c
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,9 @@ def no_backend(program: itir.FencilDefinition, *args: Any, **kwargs: Any) -> Non
),
# will use the default (embedded) execution, but input/output allocated with the provided allocator
next_tests.definitions.AllocatorId.CPU_ALLOCATOR,
next_tests.definitions.AllocatorId.GPU_ALLOCATOR,
pytest.param(
next_tests.definitions.AllocatorId.GPU_ALLOCATOR, marks=pytest.mark.requires_gpu
),
]
+ OPTIONAL_PROCESSORS,
ids=lambda p: p.short_id() if p is not None else "None",
Expand Down

0 comments on commit ca7aa6c

Please sign in to comment.