test_linkage.py

import unittest
from numba import roc


class TestLinkage(unittest.TestCase):
    def test_indirection(self):
        @roc.jit(device=True)
        def base():
            pass

        @roc.jit(device=True)
        def layer1():
            base()

        @roc.jit(device=True)
        def layer2():
            layer1()
            base()

        @roc.jit
        def kernel(a):
            layer2()

        kernel[1, 1](1)


if __name__ == '__main__':
    unittest.main()