41
41
)
42
42
from executorch .backends .qualcomm .utils .utils import (
43
43
capture_program ,
44
- dump_qnn_binary_from_pte ,
44
+ dump_context_from_pte ,
45
45
from_context_binary ,
46
46
generate_htp_compiler_spec ,
47
47
generate_multi_graph_program ,
@@ -2577,7 +2577,7 @@ def test_qnn_backend_context_extraction(self):
2577
2577
pte_path = f"{ tmp_dir } /model.pte"
2578
2578
with open (pte_path , "wb" ) as f :
2579
2579
edge_prog_mgr .write_to_file (f )
2580
- dump_qnn_binary_from_pte (pte_path )
2580
+ dump_context_from_pte (pte_path )
2581
2581
2582
2582
qnn_tool = QnnTool (tmp_dir , pte_path , sample_input )
2583
2583
qnn_tool .qnn_context_binary_generator ()
@@ -2620,7 +2620,7 @@ def test_qnn_backend_dump_context_from_pte(self):
2620
2620
pte_path = f"{ tmp_dir } /model.pte"
2621
2621
with open (pte_path , "wb" ) as f :
2622
2622
edge_prog_mgr .write_to_file (f )
2623
- dump_qnn_binary_from_pte (pte_path )
2623
+ dump_context_from_pte (pte_path )
2624
2624
2625
2625
qnn_tool = QnnTool (tmp_dir , pte_path , sample_input )
2626
2626
qnn_tool .qnn_context_binary_generator ()
@@ -2641,7 +2641,7 @@ def test_qnn_backend_dump_context_from_pte(self):
2641
2641
with open (pte_path , "wb" ) as f :
2642
2642
edge_prog_mgr .write_to_file (f )
2643
2643
2644
- dump_qnn_binary_from_pte (pte_path )
2644
+ dump_context_from_pte (pte_path )
2645
2645
binary_name = f"{ tmp_dir } /forward_0.bin"
2646
2646
self .assertTrue (os .path .isfile (binary_name ))
2647
2647
with open (binary_name , "rb" ) as f :
@@ -3225,7 +3225,7 @@ def test_qnn_backend_context_extraction(self):
3225
3225
pte_path = f"{ tmp_dir } /model.pte"
3226
3226
with open (pte_path , "wb" ) as f :
3227
3227
edge_prog_mgr .write_to_file (f )
3228
- dump_qnn_binary_from_pte (pte_path )
3228
+ dump_context_from_pte (pte_path )
3229
3229
3230
3230
qnn_tool = QnnTool (tmp_dir , pte_path , sample_input )
3231
3231
qnn_tool .qnn_context_binary_generator ()
@@ -3269,7 +3269,7 @@ def test_qnn_backend_dump_context_from_pte(self):
3269
3269
pte_path = f"{ tmp_dir } /model.pte"
3270
3270
with open (pte_path , "wb" ) as f :
3271
3271
edge_prog_mgr .write_to_file (f )
3272
- dump_qnn_binary_from_pte (pte_path )
3272
+ dump_context_from_pte (pte_path )
3273
3273
3274
3274
qnn_tool = QnnTool (tmp_dir , pte_path , sample_input )
3275
3275
qnn_tool .qnn_context_binary_generator ()
@@ -3290,7 +3290,7 @@ def test_qnn_backend_dump_context_from_pte(self):
3290
3290
with open (pte_path , "wb" ) as f :
3291
3291
edge_prog_mgr .write_to_file (f )
3292
3292
3293
- dump_qnn_binary_from_pte (pte_path )
3293
+ dump_context_from_pte (pte_path )
3294
3294
binary_name = f"{ tmp_dir } /forward_0.bin"
3295
3295
self .assertTrue (os .path .isfile (binary_name ))
3296
3296
with open (binary_name , "rb" ) as f :
0 commit comments