I have the following warning when running TVM in Ubuntu on both X86 servers. Does anyone have the same problem? Is there a solution? Thank you.
Here’s a simple piece of code I ran recently with the same warning:
import onnx import time import tvm import numpy as np import tvm.relay as relay from PIL import Image
onnx_model = onnx.load(‘mobilenetv2.onnx’)
mean = [123., 117., 104.]
std = [58.395, 57.12, 57.375]def transform_image(image):
image = image - np.array(mean) image /= np.array(std) image = np.array(image).transpose((2, 0, 1)) image = image[np.newaxis, :].astype(‘float32’) return imageimg = Image.open(’./images/cat.png’).resize((224, 224)).convert(“RGB”) x = transform_image(img) #print(x)
#-------------------------------------------------------------------------------------------------------------------
target = ‘llvm’ input_name = ‘input.1’
shape_dict = {input_name: x.shape}sym, params = relay.frontend.from_onnx(onnx_model, shape_dict)
#print(sym,params) with relay.build_config(opt_level=3): intrp = relay.build_module.create_executor(‘graph’, sym, tvm.cpu(0), target)
dtype = ‘float32’
func = intrp.evaluate()
output = func(tvm.nd.array(x.astype(dtype)), **params).asnumpy() print(output.argmax())
since = time.time() for i in range(100): output = func(tvm.nd.array(x.astype(dtype)), **params).asnumpy() time_elapsed = time.time() - since print(‘Time elapsed is {:.0f}m {:.0f}s’. format(time_elapsed // 60, time_elapsed % 60))
