diff --git a/pystencils/backends/cuda_backend.py b/pystencils/backends/cuda_backend.py index e9a78160d8ce43e4aee01ab77302749c1bd9c311..15fbde91d92c42fd604002770f23efaede31f34d 100644 --- a/pystencils/backends/cuda_backend.py +++ b/pystencils/backends/cuda_backend.py @@ -58,6 +58,7 @@ class CudaBackend(CBackend): class CudaSympyPrinter(CustomSympyPrinter): + language = "CUDA" def __init__(self): super(CudaSympyPrinter, self).__init__() diff --git a/pystencils_tests/test_cuda_known_functions.py b/pystencils_tests/test_cuda_known_functions.py index ca0d120530d069a3c7673f9328eeda6cbd23c1ad..c249c144b128c28978bc97ae15438fb6a0bda799 100644 --- a/pystencils_tests/test_cuda_known_functions.py +++ b/pystencils_tests/test_cuda_known_functions.py @@ -55,7 +55,7 @@ def test_cuda_unknown(): ast = pystencils.create_kernel(assignments, 'gpu') code = str(pystencils.show_code(ast)) print(code) - assert "Not supported" in code + assert "Not supported in CUDA" in code def main():