Skip to content

Commit bda2ec7

Browse files
authored
Fixing database bug (#18409)
* Fixing database bug * Fix lit gemini error
1 parent be37afd commit bda2ec7

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

include/tvm/meta_schedule/database.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
#include <tvm/tir/schedule/schedule.h>
3232
#include <tvm/tir/schedule/trace.h>
3333

34+
#include <filesystem>
3435
#include <memory>
3536

3637
namespace tvm {

src/relax/transform/meta_schedule.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,7 @@ Pass MetaScheduleApplyDatabase(ffi::Optional<ffi::String> work_dir, bool enable_
8686
database = Database::Current().value();
8787
} else {
8888
ICHECK(work_dir.has_value());
89+
std::filesystem::create_directories(work_dir.value().c_str());
8990
ffi::String path_workload = work_dir.value() + "/database_workload.json";
9091
ffi::String path_tuning_record = work_dir.value() + "/database_tuning_record.json";
9192
LOG(WARNING) << "Creating JSONDatabase. Workload at: " << path_workload

0 commit comments

Comments
 (0)