Home
last modified time | relevance | path

Searched refs:kernel_lib_path (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_eager/
H A Dkernel_holder.cpp444 auto kernel_lib_path = produce_aoti_kernel_lib(op, keyset, stack); in cache_miss() local
446 kernel = load_aoti_model_runner(kernel_lib_path); in cache_miss()
523 auto kernel_lib_path = py::cast<std::string>(result); in produce_aoti_kernel_lib() local
525 !kernel_lib_path.empty(), in produce_aoti_kernel_lib()
533 return kernel_lib_path; in produce_aoti_kernel_lib()
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Daoti_eager.py227 kernel_lib_path = torch._export.aot_compile(
267 Path(kernel_lib_path).relative_to(persistent_cache).as_posix()
294 return kernel_lib_path
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor.py834 kernel_lib_path = aoti_compile_with_persistent_cache(
843 self.assertTrue(Path(kernel_lib_path).exists())
877 kernel_lib_path = aoti_compile_with_persistent_cache(
886 self.assertTrue(len(kernel_lib_path) > 0)
906 self.assertTrue(kernel_lib_path in kernel_libs_abs_path)
925 kernel_lib_path = aoti_compile_with_persistent_cache(
934 self.assertTrue(Path(kernel_lib_path).exists())