File size: 184 Bytes
7e91bd7
 
 
 
 
 
 
 
 
 
671defb
1
2
3
4
5
6
7
8
9
10
11
12
# Copyright (c) FULIUCANSHENG.
# Licensed under the MIT License.

import time
import torch

tensor = torch.rand(10240, 10240).cuda()

while True:
    _ = torch.matmul(tensor, tensor)