@@ -932,7 +932,6 @@ def prepare_node(
932
932
self .fn
933
933
934
934
def make_thunk (self , node , storage_map , compute_map , no_recycling , impl = None ):
935
- from pytensor .link .c .basic import CLinker
936
935
from pytensor .link .vm import VMLinker
937
936
938
937
self .prepare_node (node , storage_map , compute_map , impl )
@@ -963,14 +962,14 @@ def thunk_wrapper(thunk=thunk, node_compute_map=node_compute_map):
963
962
return thunk_wrapper
964
963
965
964
if impl != "py" :
966
- try :
967
- # We default to CLinker because it generates code for the whole graph that the compiler can reason about.
968
- # Whereas the VMLinker will compile each node separately and call them in a pre-defined VM.
969
- # It also has less overhead
970
- return create_thunk (linker = CLinker ())
971
- except NotImplementedError :
972
- # Some Op doesn't have a C implementation, VM it is
973
- return create_thunk (VMLinker (use_cloop = True , c_thunks = True ))
965
+ # try:
966
+ # # We default to CLinker because it generates code for the whole graph that the compiler can reason about.
967
+ # # Whereas the VMLinker will compile each node separately and call them in a pre-defined VM.
968
+ # # It also has less overhead
969
+ # return create_thunk(linker=CLinker())
970
+ # except NotImplementedError:
971
+ # # Some Op doesn't have a C implementation, VM it is
972
+ return create_thunk (VMLinker (use_cloop = True , c_thunks = True ))
974
973
else :
975
974
return create_thunk (VMLinker (use_cloop = False , c_thunks = False ))
976
975
0 commit comments