Spaces:
Sleeping
Sleeping
File size: 1,024 Bytes
9060565 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
"""
Copyright © 2023 Howard Hughes Medical Institute, Authored by Carsen Stringer and Marius Pachitariu.
"""
import os, sys
os.environ["MKLDNN_VERBOSE"] = "1"
import numpy as np
import time
try:
import mxnet as mx
x = mx.sym.Variable("x")
MXNET_ENABLED = True
except:
MXNET_ENABLED = False
def test_mkl():
if MXNET_ENABLED:
num_filter = 32
kernel = (3, 3)
pad = (1, 1)
shape = (32, 32, 256, 256)
x = mx.sym.Variable("x")
w = mx.sym.Variable("w")
y = mx.sym.Convolution(data=x, weight=w, num_filter=num_filter, kernel=kernel,
no_bias=True, pad=pad)
exe = y.simple_bind(mx.cpu(), x=shape)
exe.arg_arrays[0][:] = np.random.normal(size=exe.arg_arrays[0].shape)
exe.arg_arrays[1][:] = np.random.normal(size=exe.arg_arrays[1].shape)
exe.forward(is_train=False)
o = exe.outputs[0]
t = o.asnumpy()
if __name__ == "__main__":
if MXNET_ENABLED:
test_mkl()
|