updated a test to call "use()"

tests/test_bench_loopfusion.py

@@ -254,14 +254,14 @@ class Config(object):
 
     lr = 0.001
 
-def test_bench_elemwise(n_iter=100, **kwargs):
+def test_bench_elemwise(n_iter=100, use_gpu=True, **kwargs):
     conf = Config()
     for k in kwargs:
         setattr(conf, k, kwargs[k])
 
-    if conf.use_gpu:
+    if use_gpu:
         import theano_cuda_ndarray
-        theano_cuda_ndarray.handle_shared_float32(True) 
+        theano_cuda_ndarray.use()
 
     # get symbolic train set
     s_lr = theano.tensor.fscalar()