Exemplo n.º 1
0
def ensure_supported_ccs_initialized():
    from numba.cuda import is_available as cuda_is_available
    from numba.cuda.cudadrv import nvvm

    if cuda_is_available():
        # Ensure that cudart.so is loaded and the list of supported compute
        # capabilities in the nvvm module is populated before a fork. This is
        # needed because some compilation tests don't require a CUDA context,
        # but do use NVVM, and it is required that libcudart.so should be
        # loaded before a fork (note that the requirement is not explicitly
        # documented).
        nvvm.get_supported_ccs()
Exemplo n.º 2
0
 def test_get_arch_option(self):
     # Test returning the nearest lowest arch.
     self.assertEqual(get_arch_option(5, 0), 'compute_50')
     self.assertEqual(get_arch_option(5, 1), 'compute_50')
     self.assertEqual(get_arch_option(3, 7), 'compute_35')
     # Test known arch.
     supported_cc = get_supported_ccs()
     for arch in supported_cc:
         self.assertEqual(get_arch_option(*arch), 'compute_%d%d' % arch)
     self.assertEqual(get_arch_option(1000, 0),
                      'compute_%d%d' % supported_cc[-1])
Exemplo n.º 3
0
 def test_nvvm_support(self):
     """Test supported CC by NVVM
     """
     for arch in get_supported_ccs():
         self._test_nvvm_support(arch=arch)