10 lines
315 B
Python
10 lines
315 B
Python
import torch
|
|
from torch.autograd import Variable
|
|
|
|
torch.ops.load_library("/home/philippe/Development/triton/build/examples/python/pytorch/libtorch_triton.so")
|
|
|
|
d = torch.empty(64, 64, 64, 64).uniform_(0, 1).cuda()
|
|
w = torch.empty(64, 3, 3, 64).uniform_(0, 1).cuda()
|
|
a = torch.ops.triton.conv_forward(d, w)
|
|
print(a)
|