Expand WMMA intrinsic mappings for accurate risk heatmap
Browse files
knowledge/cuda_mappings.py
CHANGED
|
@@ -349,6 +349,10 @@ HARDWARE_AWARE_MAPPINGS = {
|
|
| 349 |
"nvcuda::wmma": {
|
| 350 |
"replacement": "rocwmma",
|
| 351 |
"note": "Library Abstraction: Replaced nvcuda::wmma namespace with rocwmma."
|
|
|
|
|
|
|
|
|
|
|
|
|
| 352 |
}
|
| 353 |
}
|
| 354 |
|
|
|
|
| 349 |
"nvcuda::wmma": {
|
| 350 |
"replacement": "rocwmma",
|
| 351 |
"note": "Library Abstraction: Replaced nvcuda::wmma namespace with rocwmma."
|
| 352 |
+
},
|
| 353 |
+
"wmma::": {
|
| 354 |
+
"replacement": "rocwmma::",
|
| 355 |
+
"note": "Library Abstraction: Replaced NVIDIA wmma namespace with AMD rocwmma. Verification required."
|
| 356 |
}
|
| 357 |
}
|
| 358 |
|