comaniac commented on a change in pull request #9419:
URL: https://github.com/apache/tvm/pull/9419#discussion_r740709603
##########
File path: python/tvm/contrib/cutlass/build.py
##########
@@ -105,9 +131,19 @@ def tune_cutlass_kernels(mod, sm, profile_all=True,
use_multiprocessing=False, t
MM = arg0_shape[0]
KK = arg0_shape[1]
NN = arg1_shape[0]
- out = cutlass_profiler.profile(
- MM, NN, KK, annotator.signature["ret_dtype"], profile_all,
use_multiprocessing
- )
+ out_dtype = annotator.signature["ret_dtype"]
+ if any(isinstance(s, tvm.tir.Any) for s in [MM, KK, NN]):
+ out = cutlass_profiler.get_default(out_dtype)
+ print("Picked the default kernel " + out["name"])
Review comment:
This is because you set the logging level to INFO in the test script.
It's not trivial, unfortunately, in Python to filter logging in the way you
want. Here we could have two options:
1. Keep the current level so that users have to explicitly set the logging
level to display it.
2. Use WARNING level.
--
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]
