[RUNTIME] Support environments with multiple cudalibs (#733)

2022-10-03 19:36:24 +01:00
parent 4a2d3b7d79
commit 2b0f877fad
2 changed files with 9 additions and 7 deletions
--- a/python/test/unit/language/test_core.py
+++ b/python/test/unit/language/test_core.py
@@ -666,7 +666,7 @@ def test_atomic_cas():
    Lock = torch.zeros((1,), device='cuda', dtype=torch.int32)
    change_value[(1,)](Lock)

-    assert(Lock[0] == 1)
+    assert (Lock[0] == 1)

    # 2. only one block enters the critical section
    @triton.jit