OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00
Code Issues Actions 51 Packages Projects Releases Wiki Activity

consolidate ATen/test/dispatch_key_set_test.cpp with rest of DispatchKeySet tests (#151697)

Browse Source 
Doesn't seem to be a reason to have two test files for this.

Differential Revision: [D73274020](https://our.internmc.facebook.com/intern/diff/D73274020/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/151697
Approved by: https://github.com/Skylion007
ghstack dependencies: #151626, #151627, #151628, #151629, #151630
This commit is contained in:
Scott Wolchok
2025-04-20 09:23:28 -07:00
committed by PyTorch MergeBot
parent 9c2ac2b876
commit 8eb21dffa9
3 changed files with 10 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
c10/test/core/DispatchKeySet_test.cpp
View File

@@ -443,3 +443,13 @@ TEST(DispatchKeySet, TestFunctionalityDispatchKeyToString) {
seen_strings.insert(res);
}
}
TEST(DispatchKeySet, TestGetRuntimeDispatchKeySet) {
// Check if getRuntimeDispatchKeySet and runtimeDispatchKeySetHas agree.
for (auto dk1 : DispatchKeySet(DispatchKeySet::FULL)) {
auto dks = getRuntimeDispatchKeySet(dk1);
for (auto dk2 : DispatchKeySet(DispatchKeySet::FULL)) {
ASSERT_EQ(dks.has(dk2), runtimeDispatchKeySetHas(dk1, dk2));
}
}
}