Skip to content

Commit 48da10f

Browse files
Michael Gschwindmalfet
Michael Gschwind
authored andcommitted
add AOTI runner args comment to document arguments passed to runner creation
1 parent 6f1debc commit 48da10f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

runner-aoti/run.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -63,8 +63,8 @@ void build_transformer(Transformer *t, char* checkpoint_path, int vocab_size, in
6363
t->config.seq_len = seq_len;
6464
malloc_run_state(&t->state, &t->config);
6565
t->runner = new torch::inductor::AOTIModelContainerRunnerCpu(
66-
checkpoint_path,
67-
1
66+
/* path to model DSO */ checkpoint_path,
67+
/* thread pool size */ 1
6868
);
6969
}
7070

0 commit comments

Comments
 (0)