bharatcoder commited on
Commit
bdbbc50
·
verified ·
1 Parent(s): b4c4327

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -13,8 +13,8 @@ print(f"torch_version: {torch.__version__}")
13
  #B = np.array([[5, 6], [7, 8]])
14
 
15
  # Define large matrices
16
- A = np.random.rand(1000, 1000) # Random 1000x1000 matrix
17
- B = np.random.rand(1000, 1000)
18
 
19
  # Start the timer
20
  start_time = time.time()
@@ -34,8 +34,8 @@ print(f"Time taken for matrix multiplication with NumPy: {end_time - start_time:
34
  #B = torch.tensor([[5, 6], [7, 8]])
35
 
36
  # Define large matrices
37
- A = torch.rand(1000, 1000) # Random 1000x1000 matrix
38
- B = torch.rand(1000, 1000)
39
 
40
  # Start the timer
41
  start_time = time.time()
@@ -56,8 +56,8 @@ def zeroGPU_test(text):
56
  #B = torch.tensor([[5, 6], [7, 8]])
57
 
58
  # Define large matrices
59
- A = torch.rand(1000, 1000) # Random 1000x1000 matrix
60
- B = torch.rand(1000, 1000)
61
 
62
  # Start the timer
63
  start_time = time.time()
 
13
  #B = np.array([[5, 6], [7, 8]])
14
 
15
  # Define large matrices
16
+ A = np.random.rand(10000, 10000) # Random 10000x10000 matrix
17
+ B = np.random.rand(10000, 10000)
18
 
19
  # Start the timer
20
  start_time = time.time()
 
34
  #B = torch.tensor([[5, 6], [7, 8]])
35
 
36
  # Define large matrices
37
+ A = torch.rand(10000, 10000) # Random 10000x10000 matrix
38
+ B = torch.rand(10000, 10000)
39
 
40
  # Start the timer
41
  start_time = time.time()
 
56
  #B = torch.tensor([[5, 6], [7, 8]])
57
 
58
  # Define large matrices
59
+ A = torch.rand(10000, 10000) # Random 10000x10000 matrix
60
+ B = torch.rand(10000, 10000)
61
 
62
  # Start the timer
63
  start_time = time.time()