Skip to content

Commit f956a3b

Browse files
authored
add compute_capability [fluid_ops] (PaddlePaddle#76245)
* add compute_capability * fix
1 parent 2adc9ec commit f956a3b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddle/phi/backends/gpu/gpu_resources.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,8 @@ void InitGpuProperties(Place place,
9393
{86, "Ampere"},
9494
{89, "Ada Lovelace"},
9595
{90, "Hopper"},
96+
{100, "Blackwell"},
97+
{120, "Blackwell"},
9698
};
9799
if (!arch_computing_mapping_table.count(*compute_capability)) {
98100
LOG(ERROR) << "Mismatched GPU Architecture: "

0 commit comments

Comments
 (0)