slow_newbackend.py

import numpy as np
import theano
import theano.tensor as T
import resource

def random_matrix_mult():
	mat1 = T.fmatrix('mat1')
	mat2 = T.fmatrix('mat2')
	col_index = np.arange(0, 150000, 5000)
	row_index = np.arange(10000)
	first_matrix = col_index[:,np.newaxis] + row_index[np.newaxis, :]
	second_matrix = np.random.random((20, 10000))
	dot_output = T.dot(mat1, mat2.T)
	dot_func = theano.function([mat1, mat2], dot_output, allow_input_downcast=True)
	prior = resource.getrusage(resource.RUSAGE_SELF)
	output = dot_func(first_matrix, second_matrix)
	post = resource.getrusage(resource.RUSAGE_SELF)
	print("User time for matrix theano: " + str(post.ru_utime - prior.ru_utime))

if __name__ == '__main__':
	for i in range(2):
		random_matrix_mult()