Skip to content

[Notebook 5] tune_tir is not working. API might be changed. #35

@wonseokjeon

Description

@wonseokjeon
from tvm import meta_schedule as ms

database = ms.tune_tir(
    mod=MyModule,
    target="llvm --num-cores=1",
    max_trials_global=64,
    num_trials_per_iter=64,
    space=ms.space_generator.ScheduleFn(stochastic_schedule_mm),
    work_dir="./tune_tmp",
    task_name="main"
)

When I was trying to run this block, It gave an error "TypeError: tune_tir() got an unexpected keyword argument 'task_name'".
API might be changed.

So I'm trying to fix this by,

from tvm import meta_schedule as ms

database = ms.tune_tir(
    mod=MyModule,
    target="llvm --num-cores=1",
    max_trials_global=64,
    num_trials_per_iter=64,
    space=ms.space_generator.ScheduleFn(stochastic_schedule_mm),
    work_dir="./tune_tmp",
)

Then the error "ValueError: TensorIntrin 'dot_4x4_i8i8s32_neon' is not registered" came out.
I'm running this notebook on M1 Max Macbook Pro, python 3.10.

How can i fix it? Did anyone resolve this issue? Pleas give me some hints.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions