github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[BACKEND] Now using call_once to initialize LLVM target (#1373)

Browse Source
This commit is contained in:
Philippe Tillet
2023-03-19 21:23:39 -07:00
committed by GitHub
parent 6366c5a254
commit b4decbe155
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/test/regression/test_performance.py
View File

@@ -162,7 +162,7 @@ def test_elementwise(N):
flash_attention_data = {
"a100": {
(4, 48, 4096, 64, 'forward', 'float16'): 0.37,
(4, 48, 4096, 64, 'backward', 'float16'): 0.25,
(4, 48, 4096, 64, 'backward', 'float16'): 0.26,
}
}