Skip to content

Commit 8bad072

Browse files
authored
Change device retrieval method to use torch.device because Triton 3.0 API no longer supports the device interface. (#402)
1 parent 438bc67 commit 8bad072

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

kernels/openai-triton/vector-add/triton_vector_add.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,7 @@
2424
import triton
2525
import triton.language as tl
2626

27-
DEVICE = triton.runtime.driver.active.get_active_torch_device()
28-
27+
DEVICE = torch.device("cuda:0")
2928

3029
@triton.jit
3130
def add_kernel(

0 commit comments

Comments
 (0)