Skip to content

Commit c38297f

Browse files
committed
[SW] Forced torch device to cpu in sw benchmarks
1 parent 180641d commit c38297f

5 files changed

Lines changed: 5 additions & 5 deletions

File tree

sw/spatzBenchmarks/hp-fmatmul/script/gen_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
np.random.seed(42)
1616
torch.manual_seed(42)
1717

18-
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
18+
device = torch.device("cpu")
1919

2020
global verbose
2121

sw/spatzBenchmarks/sdotp-bp-fmatmul/script/gen_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
np.random.seed(42)
1616
torch.manual_seed(42)
1717

18-
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
18+
device = torch.device("cpu")
1919

2020
global verbose
2121

sw/spatzBenchmarks/sdotp-hp-fmatmul/script/gen_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
np.random.seed(42)
1616
torch.manual_seed(42)
1717

18-
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
18+
device = torch.device("cpu")
1919

2020
global verbose
2121

sw/spatzBenchmarks/widening-bp-fmatmul/script/gen_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
np.random.seed(42)
1616
torch.manual_seed(42)
1717

18-
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
18+
device = torch.device("cpu")
1919

2020
global verbose
2121

sw/spatzBenchmarks/widening-hp-fmatmul/script/gen_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
np.random.seed(42)
1616
torch.manual_seed(42)
1717

18-
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
18+
device = torch.device("cpu")
1919

2020
global verbose
2121

0 commit comments

Comments
 (0)