frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Python Dispatcher integration with C++ dispatcher (#85050)

Browse Source 
#84826 but without ghstack
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85050
Approved by: https://github.com/malfet
This commit is contained in:
Michael Voznesensky
2022-09-15 00:43:36 +00:00
committed by PyTorch MergeBot
parent 3a107bc9be
commit 8ca1839d32
33 changed files with 685 additions and 161 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_prims.py
View File

@ -327,7 +327,7 @@ class TestPrims(TestCase):
# Check that all call_function nodes are prims
call_function_nodes = list(filter(lambda n: n.op == "call_function", gm.graph.nodes))
all_prims_namespace = all(
node.target.name.startswith("prims") for node in call_function_nodes
node.target.name().startswith("prims") for node in call_function_nodes
)
self.assertTrue(all_prims_namespace)