We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d392d25 commit da2e89aCopy full SHA for da2e89a
src/meta_schedule/feature_extractor/per_store_feature.cc
@@ -1392,7 +1392,8 @@ class PerStoreFeatureNode : public FeatureExtractorNode {
1392
1393
Array<runtime::NDArray> ExtractFrom(const TuneContext& tune_context,
1394
const Array<MeasureCandidate>& candidates) {
1395
- bool is_gpu = tune_context->target.value()->kind->name == "cuda";
+ auto& target_keys = tune_context->target.value()->keys;
1396
+ bool is_gpu = std::find(target_keys.begin(), target_keys.end(), "gpu") != target_keys.end();
1397
std::vector<runtime::NDArray> results;
1398
results.resize(candidates.size());
1399
std::unique_ptr<tir::group6::Feature> feature_group6 = nullptr;
0 commit comments