iwilldoit commited on
Commit
68bbd40
·
1 Parent(s): fb42210

Expand WMMA intrinsic mappings for accurate risk heatmap

Browse files
Files changed (1) hide show
  1. knowledge/cuda_mappings.py +4 -0
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