Open nvukobratTT opened 1 week ago
To confirm that each op individually works for the Llama model, let's add a set of op tests that are specific to this variation of the Llama model.
To do so, here are a few hints:
forge/test/mlir/llama/ops/*
compiler_cfg.compile_depth = forge.CompileDepth.SPLIT_GRAPH
Additional info for each of the ops that doesnt work:
Summary
To confirm that each op individually works for the Llama model, let's add a set of op tests that are specific to this variation of the Llama model.
To do so, here are a few hints:
forge/test/mlir/llama/ops/*
compiler_cfg.compile_depth = forge.CompileDepth.SPLIT_GRAPH
)