We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 438bc67 commit 8bad072Copy full SHA for 8bad072
1 file changed
kernels/openai-triton/vector-add/triton_vector_add.py
@@ -24,8 +24,7 @@
24
import triton
25
import triton.language as tl
26
27
-DEVICE = triton.runtime.driver.active.get_active_torch_device()
28
-
+DEVICE = torch.device("cuda:0")
29
30
@triton.jit
31
def add_kernel(
0 commit comments