2019-05-03 14:30:06 -04:00
|
|
|
import torch
|
|
|
|
|
from torch.autograd import Variable
|
|
|
|
|
|
2019-05-04 01:32:34 -04:00
|
|
|
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)
|
