AndrewZhaoLuo commented on code in PR #11066:
URL: https://github.com/apache/tvm/pull/11066#discussion_r858242458
##########
tests/python/unittest/test_runtime_profiling.py:
##########
@@ -257,6 +259,50 @@ def test_profile_function(target, dev):
assert report[metric].value > 0
[email protected]_targets("llvm")
+def test_estimate_peak_fma_flops(target, dev):
+ # This test uses vectorized instructions so we need a target that supports
them
+ if target == "llvm":
+ target = "llvm -mattr=+fma,+avx2"
Review Comment:
Yeah so this runs fine on m1 mac. I get around 2e11 FLOPS.
However this is with just 'llvm' target reaching in which is interesting.
Changing the target string to any of the following does not change results, at
least not noticeably. It does output some things
"llvm"
"llvm -mcpu=apple-latest -mtriple=arm64-apple-macos"
Additionally
"llvm -mattr=+fma,+avx2" generates the following printout
```
tests/python/unittest/test_runtime_profiling.py '+fma' is not a recognized
feature for this target (ignoring feature)
'+avx2' is not a recognized feature for this target (ignoring feature)
'+fma' is not a recognized feature for this target (ignoring feature)
'+avx2' is not a recognized feature for this target (ignoring feature)
'+fma' is not a recognized feature for this target (ignoring feature)
'+avx2' is not a recognized feature for this target (ignoring feature)
'+fma' is not a recognized feature for this target (ignoring feature)
'+avx2' is not a recognized feature for this target (ignoring feature)
'+fma' is not a recognized feature for this target (ignoring feature)
'+avx2' is not a recognized feature for this target (ignoring feature)
'+fma' is not a recognized feature for this target (ignoring feature)
'+avx2' is not a recognized feature for this target (ignoring feature)
```
Can't grep for the string above so Imma assume this is just llvm carrying
TVM somehow
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
