()
| 52 | } |
| 53 | |
| 54 | def cuda_malloc_supported(): |
| 55 | try: |
| 56 | names = get_gpu_names() |
| 57 | except: |
| 58 | names = set() |
| 59 | for x in names: |
| 60 | if "NVIDIA" in x: |
| 61 | for b in blacklist: |
| 62 | if b in x: |
| 63 | return False |
| 64 | return True |
| 65 | |
| 66 | |
| 67 | version = "" |
no test coverage detected