maochen:

import torch rand1 = torch.rand((2, 256)).cuda() rand2 = torch.rand((2, 256)).cuda() c_x = torch.mm(rand1, rand2.t()) c_x_output = [rand1, rand2]


The workground is to run the following code after import torch:
>>> import ctypes
>>> ctypes.cdll.LoadLibrary('caffe2_nvrtc.dll')