diff --git a/source/gpu.py b/source/gpu.py index 13d1ed3..4837c3f 100644 --- a/source/gpu.py +++ b/source/gpu.py @@ -45,7 +45,7 @@ def data_shipping_experiment(n:int): array2 = np.random.randn(200,200) t0 = time.time() for i in range(n): - array3 = array1.matmul(array2) + array3 = array1.dot(array2) array1 = array3 t1 = time.time() @@ -65,7 +65,7 @@ def data_shipping_experiment(n:int): print(f'GPU only operations took {t1-t0}') - #let's ship data like a mofo + #let's a lot of data tensor1 = torch.FloatTensor(200, 200) tensor2 = torch.FloatTensor(200, 200) @@ -94,4 +94,4 @@ def data_shipping_experiment(n:int): # view_number(_data.train_data[10], str(_data.train_labels[10])) data_shipping_experiment(100000) - this_wont_work_dummy(_data.train_data, _data.train_labels) \ No newline at end of file + this_wont_work_dummy(_data.train_data, _data.train_labels)