Skip to content
On PR

TVM cleanup #2778

Sign in to view logs

TVM cleanup

TVM cleanup #2778

GitHub Actions / TT-Forge-FE Tests failed Feb 26, 2025 in 0s

672 tests run, 464 passed, 136 skipped, 72 failed.

Annotations

Check failure on line 117 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_less[shape_x1-shape_y1] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape_x = (1, 64, 28, 28), shape_y = (1, 64, 28, 28)

    @pytest.mark.parametrize(
        "shape_x, shape_y",
        [
            ((1, 128, 28, 28), (1, 128, 28, 28)),
            ((1, 64, 28, 28), (1, 64, 28, 28)),
            ((1, 256, 28, 28), (1, 256, 28, 28)),
            ((1, 128, 14, 14), (1, 128, 14, 14)),
            ((1, 128, 56, 56), (1, 128, 56, 56)),
            ((1, 32, 64, 64), (1, 32, 64, 64)),
            ((1, 512, 7, 7), (1, 512, 7, 7)),
            ((1, 32, 32, 32), (1, 32, 32, 32)),
        ],
    )
    @pytest.mark.push
    def test_less(shape_x, shape_y):
        class Less(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.less(x, y)
    
        x = torch.rand(shape_x)
        y = torch.rand(shape_y)
    
        inputs = [x, y]
    
        framework_model = Less()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:117: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3ef0c58250>
pre = CallNode(Op(less), [Var(x, ty=TensorType([1, 64, 28, 28], float32)), Var(y, ty=TensorType([1, 64, 28, 28], float32))], (nullptr), [TensorType([1, 64, 28, 28], float32), TensorType([1, 64, 28, 28], float32)])
post = CallNode(Op(less), [Var(x, ty=TensorType([1, 64, 28, 28], float32)), Var(y, ty=TensorType([1, 64, 28, 28], float32))], (nullptr), [TensorType([1, 64, 28, 28], float32), TensorType([1, 64, 28, 28], float32)])
node_map = {relay.dataflow_pattern.WildcardPattern(0x55a8fdd60a50): [Var(x, ty=TensorType([1, 64, 28, 28], float32))], relay.data...[1, 64, 28, 28], float32))], (nullptr), [TensorType([1, 64, 28, 28], float32), TensorType([1, 64, 28, 28], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError

Check failure on line 117 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_less[shape_x3-shape_y3] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape_x = (1, 128, 14, 14), shape_y = (1, 128, 14, 14)

    @pytest.mark.parametrize(
        "shape_x, shape_y",
        [
            ((1, 128, 28, 28), (1, 128, 28, 28)),
            ((1, 64, 28, 28), (1, 64, 28, 28)),
            ((1, 256, 28, 28), (1, 256, 28, 28)),
            ((1, 128, 14, 14), (1, 128, 14, 14)),
            ((1, 128, 56, 56), (1, 128, 56, 56)),
            ((1, 32, 64, 64), (1, 32, 64, 64)),
            ((1, 512, 7, 7), (1, 512, 7, 7)),
            ((1, 32, 32, 32), (1, 32, 32, 32)),
        ],
    )
    @pytest.mark.push
    def test_less(shape_x, shape_y):
        class Less(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.less(x, y)
    
        x = torch.rand(shape_x)
        y = torch.rand(shape_y)
    
        inputs = [x, y]
    
        framework_model = Less()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:117: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3ef8b81c30>
pre = CallNode(Op(less), [Var(x, ty=TensorType([1, 128, 14, 14], float32)), Var(y, ty=TensorType([1, 128, 14, 14], float32))], (nullptr), [TensorType([1, 128, 14, 14], float32), TensorType([1, 128, 14, 14], float32)])
post = CallNode(Op(less), [Var(x, ty=TensorType([1, 128, 14, 14], float32)), Var(y, ty=TensorType([1, 128, 14, 14], float32))], (nullptr), [TensorType([1, 128, 14, 14], float32), TensorType([1, 128, 14, 14], float32)])
node_map = {relay.dataflow_pattern.WildcardPattern(0x55a8fcbb9e00): [Var(y, ty=TensorType([1, 128, 14, 14], float32))], relay.dat... 128, 14, 14], float32))], (nullptr), [TensorType([1, 128, 14, 14], float32), TensorType([1, 128, 14, 14], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError

Check failure on line 117 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_less[shape_x5-shape_y5] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape_x = (1, 32, 64, 64), shape_y = (1, 32, 64, 64)

    @pytest.mark.parametrize(
        "shape_x, shape_y",
        [
            ((1, 128, 28, 28), (1, 128, 28, 28)),
            ((1, 64, 28, 28), (1, 64, 28, 28)),
            ((1, 256, 28, 28), (1, 256, 28, 28)),
            ((1, 128, 14, 14), (1, 128, 14, 14)),
            ((1, 128, 56, 56), (1, 128, 56, 56)),
            ((1, 32, 64, 64), (1, 32, 64, 64)),
            ((1, 512, 7, 7), (1, 512, 7, 7)),
            ((1, 32, 32, 32), (1, 32, 32, 32)),
        ],
    )
    @pytest.mark.push
    def test_less(shape_x, shape_y):
        class Less(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.less(x, y)
    
        x = torch.rand(shape_x)
        y = torch.rand(shape_y)
    
        inputs = [x, y]
    
        framework_model = Less()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:117: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3ef8a19c00>
pre = CallNode(Op(less), [Var(x, ty=TensorType([1, 32, 64, 64], float32)), Var(y, ty=TensorType([1, 32, 64, 64], float32))], (nullptr), [TensorType([1, 32, 64, 64], float32), TensorType([1, 32, 64, 64], float32)])
post = CallNode(Op(less), [Var(x, ty=TensorType([1, 32, 64, 64], float32)), Var(y, ty=TensorType([1, 32, 64, 64], float32))], (nullptr), [TensorType([1, 32, 64, 64], float32), TensorType([1, 32, 64, 64], float32)])
node_map = {relay.dataflow_pattern.WildcardPattern(0x55a8fca5a860): [Var(x, ty=TensorType([1, 32, 64, 64], float32))], relay.data...[1, 32, 64, 64], float32))], (nullptr), [TensorType([1, 32, 64, 64], float32), TensorType([1, 32, 64, 64], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError

Check failure on line 117 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_less[shape_x7-shape_y7] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape_x = (1, 32, 32, 32), shape_y = (1, 32, 32, 32)

    @pytest.mark.parametrize(
        "shape_x, shape_y",
        [
            ((1, 128, 28, 28), (1, 128, 28, 28)),
            ((1, 64, 28, 28), (1, 64, 28, 28)),
            ((1, 256, 28, 28), (1, 256, 28, 28)),
            ((1, 128, 14, 14), (1, 128, 14, 14)),
            ((1, 128, 56, 56), (1, 128, 56, 56)),
            ((1, 32, 64, 64), (1, 32, 64, 64)),
            ((1, 512, 7, 7), (1, 512, 7, 7)),
            ((1, 32, 32, 32), (1, 32, 32, 32)),
        ],
    )
    @pytest.mark.push
    def test_less(shape_x, shape_y):
        class Less(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.less(x, y)
    
        x = torch.rand(shape_x)
        y = torch.rand(shape_y)
    
        inputs = [x, y]
    
        framework_model = Less()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:117: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3ef0b64310>
pre = CallNode(Op(less), [Var(x, ty=TensorType([1, 32, 32, 32], float32)), Var(y, ty=TensorType([1, 32, 32, 32], float32))], (nullptr), [TensorType([1, 32, 32, 32], float32), TensorType([1, 32, 32, 32], float32)])
post = CallNode(Op(less), [Var(x, ty=TensorType([1, 32, 32, 32], float32)), Var(y, ty=TensorType([1, 32, 32, 32], float32))], (nullptr), [TensorType([1, 32, 32, 32], float32), TensorType([1, 32, 32, 32], float32)])
node_map = {relay.dataflow_pattern.WildcardPattern(0x55a8fde3bb90): [Var(y, ty=TensorType([1, 32, 32, 32], float32))], relay.data...float32)])], relay.dataflow_pattern.WildcardPattern(0x55a8fde32d30): [Var(x, ty=TensorType([1, 32, 32, 32], float32))]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError

Check failure on line 150 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_greater[shape_x1-shape_y1] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape_x = (1, 64, 28, 28), shape_y = (1, 64, 28, 28)

    @pytest.mark.parametrize(
        "shape_x, shape_y",
        [
            ((1, 128, 28, 28), (1, 128, 28, 28)),
            ((1, 64, 28, 28), (1, 64, 28, 28)),
            ((1, 256, 28, 28), (1, 256, 28, 28)),
            ((1, 128, 14, 14), (1, 128, 14, 14)),
            ((1, 128, 56, 56), (1, 128, 56, 56)),
            ((1, 32, 64, 64), (1, 32, 64, 64)),
            ((1, 512, 7, 7), (1, 512, 7, 7)),
            ((1, 32, 32, 32), (1, 32, 32, 32)),
        ],
    )
    @pytest.mark.push
    def test_greater(shape_x, shape_y):
        class Greater(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.greater(x, y)
    
        x = torch.rand(shape_x)
        y = torch.rand(shape_y)
    
        inputs = [x, y]
    
        framework_model = Greater()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:150: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3ef8c1e1a0>
pre = CallNode(Op(greater), [Var(x, ty=TensorType([1, 64, 28, 28], float32)), Var(y, ty=TensorType([1, 64, 28, 28], float32))], (nullptr), [TensorType([1, 64, 28, 28], float32), TensorType([1, 64, 28, 28], float32)])
post = CallNode(Op(greater), [Var(x, ty=TensorType([1, 64, 28, 28], float32)), Var(y, ty=TensorType([1, 64, 28, 28], float32))], (nullptr), [TensorType([1, 64, 28, 28], float32), TensorType([1, 64, 28, 28], float32)])
node_map = {relay.dataflow_pattern.AltPattern(0x55a8fd06b170): [CallNode(Op(greater), [Var(x, ty=TensorType([1, 64, 28, 28], floa...: [Var(x, ty=TensorType([1, 64, 28, 28], float32))], relay.dataflow_pattern.ExprPattern(0x55a8fcc9be60): [Op(greater)]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError

Check failure on line 150 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_greater[shape_x3-shape_y3] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape_x = (1, 128, 14, 14), shape_y = (1, 128, 14, 14)

    @pytest.mark.parametrize(
        "shape_x, shape_y",
        [
            ((1, 128, 28, 28), (1, 128, 28, 28)),
            ((1, 64, 28, 28), (1, 64, 28, 28)),
            ((1, 256, 28, 28), (1, 256, 28, 28)),
            ((1, 128, 14, 14), (1, 128, 14, 14)),
            ((1, 128, 56, 56), (1, 128, 56, 56)),
            ((1, 32, 64, 64), (1, 32, 64, 64)),
            ((1, 512, 7, 7), (1, 512, 7, 7)),
            ((1, 32, 32, 32), (1, 32, 32, 32)),
        ],
    )
    @pytest.mark.push
    def test_greater(shape_x, shape_y):
        class Greater(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.greater(x, y)
    
        x = torch.rand(shape_x)
        y = torch.rand(shape_y)
    
        inputs = [x, y]
    
        framework_model = Greater()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:150: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3ef8a2ba30>
pre = CallNode(Op(greater), [Var(x, ty=TensorType([1, 128, 14, 14], float32)), Var(y, ty=TensorType([1, 128, 14, 14], float32))], (nullptr), [TensorType([1, 128, 14, 14], float32), TensorType([1, 128, 14, 14], float32)])
post = CallNode(Op(greater), [Var(x, ty=TensorType([1, 128, 14, 14], float32)), Var(y, ty=TensorType([1, 128, 14, 14], float32))], (nullptr), [TensorType([1, 128, 14, 14], float32), TensorType([1, 128, 14, 14], float32)])
node_map = {relay.dataflow_pattern.AltPattern(0x55a8fd01c830): [CallNode(Op(greater), [Var(x, ty=TensorType([1, 128, 14, 14], flo...loat32)])], relay.dataflow_pattern.WildcardPattern(0x55a8fd071ed0): [Var(x, ty=TensorType([1, 128, 14, 14], float32))]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError

Check failure on line 150 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_greater[shape_x5-shape_y5] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape_x = (1, 32, 64, 64), shape_y = (1, 32, 64, 64)

    @pytest.mark.parametrize(
        "shape_x, shape_y",
        [
            ((1, 128, 28, 28), (1, 128, 28, 28)),
            ((1, 64, 28, 28), (1, 64, 28, 28)),
            ((1, 256, 28, 28), (1, 256, 28, 28)),
            ((1, 128, 14, 14), (1, 128, 14, 14)),
            ((1, 128, 56, 56), (1, 128, 56, 56)),
            ((1, 32, 64, 64), (1, 32, 64, 64)),
            ((1, 512, 7, 7), (1, 512, 7, 7)),
            ((1, 32, 32, 32), (1, 32, 32, 32)),
        ],
    )
    @pytest.mark.push
    def test_greater(shape_x, shape_y):
        class Greater(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.greater(x, y)
    
        x = torch.rand(shape_x)
        y = torch.rand(shape_y)
    
        inputs = [x, y]
    
        framework_model = Greater()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:150: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3ef883c1c0>
pre = CallNode(Op(greater), [Var(x, ty=TensorType([1, 32, 64, 64], float32)), Var(y, ty=TensorType([1, 32, 64, 64], float32))], (nullptr), [TensorType([1, 32, 64, 64], float32), TensorType([1, 32, 64, 64], float32)])
post = CallNode(Op(greater), [Var(x, ty=TensorType([1, 32, 64, 64], float32)), Var(y, ty=TensorType([1, 32, 64, 64], float32))], (nullptr), [TensorType([1, 32, 64, 64], float32), TensorType([1, 32, 64, 64], float32)])
node_map = {relay.dataflow_pattern.AltPattern(0x55a8fcc999e0): [CallNode(Op(greater), [Var(x, ty=TensorType([1, 32, 64, 64], floa...[1, 32, 64, 64], float32))], (nullptr), [TensorType([1, 32, 64, 64], float32), TensorType([1, 32, 64, 64], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError

Check failure on line 150 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_greater[shape_x7-shape_y7] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape_x = (1, 32, 32, 32), shape_y = (1, 32, 32, 32)

    @pytest.mark.parametrize(
        "shape_x, shape_y",
        [
            ((1, 128, 28, 28), (1, 128, 28, 28)),
            ((1, 64, 28, 28), (1, 64, 28, 28)),
            ((1, 256, 28, 28), (1, 256, 28, 28)),
            ((1, 128, 14, 14), (1, 128, 14, 14)),
            ((1, 128, 56, 56), (1, 128, 56, 56)),
            ((1, 32, 64, 64), (1, 32, 64, 64)),
            ((1, 512, 7, 7), (1, 512, 7, 7)),
            ((1, 32, 32, 32), (1, 32, 32, 32)),
        ],
    )
    @pytest.mark.push
    def test_greater(shape_x, shape_y):
        class Greater(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.greater(x, y)
    
        x = torch.rand(shape_x)
        y = torch.rand(shape_y)
    
        inputs = [x, y]
    
        framework_model = Greater()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:150: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3ef88b3d30>
pre = CallNode(Op(greater), [Var(x, ty=TensorType([1, 32, 32, 32], float32)), Var(y, ty=TensorType([1, 32, 32, 32], float32))], (nullptr), [TensorType([1, 32, 32, 32], float32), TensorType([1, 32, 32, 32], float32)])
post = CallNode(Op(greater), [Var(x, ty=TensorType([1, 32, 32, 32], float32)), Var(y, ty=TensorType([1, 32, 32, 32], float32))], (nullptr), [TensorType([1, 32, 32, 32], float32), TensorType([1, 32, 32, 32], float32)])
node_map = {relay.dataflow_pattern.CallPattern(0x55a8fd053fd0): [CallNode(Op(greater), [Var(x, ty=TensorType([1, 32, 32, 32], flo...[1, 32, 32, 32], float32))], (nullptr), [TensorType([1, 32, 32, 32], float32), TensorType([1, 32, 32, 32], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError

Check failure on line 183 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_not_equal[shape_x1-shape_y1] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape_x = (1, 64, 28, 28), shape_y = (1, 64, 28, 28)

    @pytest.mark.parametrize(
        "shape_x, shape_y",
        [
            ((1, 128, 28, 28), (1, 128, 28, 28)),
            ((1, 64, 28, 28), (1, 64, 28, 28)),
            ((1, 256, 28, 28), (1, 256, 28, 28)),
            ((1, 128, 14, 14), (1, 128, 14, 14)),
            ((1, 128, 56, 56), (1, 128, 56, 56)),
            ((1, 32, 64, 64), (1, 32, 64, 64)),
            ((1, 512, 7, 7), (1, 512, 7, 7)),
            ((1, 32, 32, 32), (1, 32, 32, 32)),
        ],
    )
    @pytest.mark.push
    def test_not_equal(shape_x, shape_y):
        class NotEqual(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.ne(x, y)
    
        x = torch.rand(shape_x)
        y = torch.rand(shape_y)
    
        inputs = [x, y]
    
        framework_model = NotEqual()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:183: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3f1c1253f0>
pre = CallNode(Op(not_equal), [Var(x, ty=TensorType([1, 64, 28, 28], float32)), Var(y, ty=TensorType([1, 64, 28, 28], float32))], (nullptr), [TensorType([1, 64, 28, 28], float32), TensorType([1, 64, 28, 28], float32)])
post = CallNode(Op(not_equal), [Var(x, ty=TensorType([1, 64, 28, 28], float32)), Var(y, ty=TensorType([1, 64, 28, 28], float32))], (nullptr), [TensorType([1, 64, 28, 28], float32), TensorType([1, 64, 28, 28], float32)])
node_map = {relay.dataflow_pattern.AltPattern(0x55a8fdcdfaa0): [CallNode(Op(not_equal), [Var(x, ty=TensorType([1, 64, 28, 28], fl...[1, 64, 28, 28], float32))], (nullptr), [TensorType([1, 64, 28, 28], float32), TensorType([1, 64, 28, 28], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError

Check failure on line 183 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_not_equal[shape_x3-shape_y3] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape_x = (1, 128, 14, 14), shape_y = (1, 128, 14, 14)

    @pytest.mark.parametrize(
        "shape_x, shape_y",
        [
            ((1, 128, 28, 28), (1, 128, 28, 28)),
            ((1, 64, 28, 28), (1, 64, 28, 28)),
            ((1, 256, 28, 28), (1, 256, 28, 28)),
            ((1, 128, 14, 14), (1, 128, 14, 14)),
            ((1, 128, 56, 56), (1, 128, 56, 56)),
            ((1, 32, 64, 64), (1, 32, 64, 64)),
            ((1, 512, 7, 7), (1, 512, 7, 7)),
            ((1, 32, 32, 32), (1, 32, 32, 32)),
        ],
    )
    @pytest.mark.push
    def test_not_equal(shape_x, shape_y):
        class NotEqual(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.ne(x, y)
    
        x = torch.rand(shape_x)
        y = torch.rand(shape_y)
    
        inputs = [x, y]
    
        framework_model = NotEqual()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:183: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3ef8788d00>
pre = CallNode(Op(not_equal), [Var(x, ty=TensorType([1, 128, 14, 14], float32)), Var(y, ty=TensorType([1, 128, 14, 14], float32))], (nullptr), [TensorType([1, 128, 14, 14], float32), TensorType([1, 128, 14, 14], float32)])
post = CallNode(Op(not_equal), [Var(x, ty=TensorType([1, 128, 14, 14], float32)), Var(y, ty=TensorType([1, 128, 14, 14], float32))], (nullptr), [TensorType([1, 128, 14, 14], float32), TensorType([1, 128, 14, 14], float32)])
node_map = {relay.dataflow_pattern.CallPattern(0x55a8fcbe61b0): [CallNode(Op(not_equal), [Var(x, ty=TensorType([1, 128, 14, 14], ... 128, 14, 14], float32))], (nullptr), [TensorType([1, 128, 14, 14], float32), TensorType([1, 128, 14, 14], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError

Check failure on line 183 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_not_equal[shape_x5-shape_y5] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape_x = (1, 32, 64, 64), shape_y = (1, 32, 64, 64)

    @pytest.mark.parametrize(
        "shape_x, shape_y",
        [
            ((1, 128, 28, 28), (1, 128, 28, 28)),
            ((1, 64, 28, 28), (1, 64, 28, 28)),
            ((1, 256, 28, 28), (1, 256, 28, 28)),
            ((1, 128, 14, 14), (1, 128, 14, 14)),
            ((1, 128, 56, 56), (1, 128, 56, 56)),
            ((1, 32, 64, 64), (1, 32, 64, 64)),
            ((1, 512, 7, 7), (1, 512, 7, 7)),
            ((1, 32, 32, 32), (1, 32, 32, 32)),
        ],
    )
    @pytest.mark.push
    def test_not_equal(shape_x, shape_y):
        class NotEqual(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.ne(x, y)
    
        x = torch.rand(shape_x)
        y = torch.rand(shape_y)
    
        inputs = [x, y]
    
        framework_model = NotEqual()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:183: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3ef89a2890>
pre = CallNode(Op(not_equal), [Var(x, ty=TensorType([1, 32, 64, 64], float32)), Var(y, ty=TensorType([1, 32, 64, 64], float32))], (nullptr), [TensorType([1, 32, 64, 64], float32), TensorType([1, 32, 64, 64], float32)])
post = CallNode(Op(not_equal), [Var(x, ty=TensorType([1, 32, 64, 64], float32)), Var(y, ty=TensorType([1, 32, 64, 64], float32))], (nullptr), [TensorType([1, 32, 64, 64], float32), TensorType([1, 32, 64, 64], float32)])
node_map = {relay.dataflow_pattern.WildcardPattern(0x55a8fc9e4df0): [Var(y, ty=TensorType([1, 32, 64, 64], float32))], relay.data...[1, 32, 64, 64], float32))], (nullptr), [TensorType([1, 32, 64, 64], float32), TensorType([1, 32, 64, 64], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError

Check failure on line 183 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_not_equal[shape_x7-shape_y7] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape_x = (1, 32, 32, 32), shape_y = (1, 32, 32, 32)

    @pytest.mark.parametrize(
        "shape_x, shape_y",
        [
            ((1, 128, 28, 28), (1, 128, 28, 28)),
            ((1, 64, 28, 28), (1, 64, 28, 28)),
            ((1, 256, 28, 28), (1, 256, 28, 28)),
            ((1, 128, 14, 14), (1, 128, 14, 14)),
            ((1, 128, 56, 56), (1, 128, 56, 56)),
            ((1, 32, 64, 64), (1, 32, 64, 64)),
            ((1, 512, 7, 7), (1, 512, 7, 7)),
            ((1, 32, 32, 32), (1, 32, 32, 32)),
        ],
    )
    @pytest.mark.push
    def test_not_equal(shape_x, shape_y):
        class NotEqual(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.ne(x, y)
    
        x = torch.rand(shape_x)
        y = torch.rand(shape_y)
    
        inputs = [x, y]
    
        framework_model = NotEqual()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:183: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3ef8bb2cb0>
pre = CallNode(Op(not_equal), [Var(x, ty=TensorType([1, 32, 32, 32], float32)), Var(y, ty=TensorType([1, 32, 32, 32], float32))], (nullptr), [TensorType([1, 32, 32, 32], float32), TensorType([1, 32, 32, 32], float32)])
post = CallNode(Op(not_equal), [Var(x, ty=TensorType([1, 32, 32, 32], float32)), Var(y, ty=TensorType([1, 32, 32, 32], float32))], (nullptr), [TensorType([1, 32, 32, 32], float32), TensorType([1, 32, 32, 32], float32)])
node_map = {relay.dataflow_pattern.AltPattern(0x55a8fdcfa050): [CallNode(Op(not_equal), [Var(x, ty=TensorType([1, 32, 32, 32], fl...float32)])], relay.dataflow_pattern.WildcardPattern(0x55a8fdd48bf0): [Var(x, ty=TensorType([1, 32, 32, 32], float32))]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError

Check failure on line 232 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_equal[shape0] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape = (1, 128, 28, 28)

    @pytest.mark.parametrize(
        "shape",
        [
            (1, 128, 28, 28),
            (1, 64, 28, 28),
            (1, 256, 28, 28),
            (1, 128, 14, 14),
            (1, 128, 56, 56),
            (1, 32, 64, 64),
            (1, 512, 7, 7),
            (1, 32, 32, 32),
            (128, 28, 28),
            (64, 28, 28),
            (256, 28, 28),
            (128, 14, 14),
            (128, 56, 56),
            (32, 64, 64),
            (512, 7, 7),
            (32, 32, 32),
            (128, 28),
            (64, 28),
            (256, 28),
            (128, 14),
            (128, 56),
            (32, 64),
            (512, 7),
            (32, 32),
        ],
    )
    @pytest.mark.push
    def test_equal(shape):
        class Equal(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.eq(x, y)
    
        x = torch.rand(shape)
        y = x * 2.0
    
        inputs = [x, y]
    
        framework_model = Equal()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:232: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1095: in compile_for_forge
    compiled_relay_module = run_forge_compile_passes(relay_module, params, inputs, target, framework_outputs, verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4238: in run_forge_compile_passes
    return run_pattern_callbacks(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4229: in run_pattern_callbacks
    raise ex
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4226: in run_pattern_callbacks
    relay_module = _run_pattern_callback(relay_module, callback, callback_name)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4205: in _run_pattern_callback
    relay_module['main'] = rewrite(callback, relay_module['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge_passes.InverseMaskGen object at 0x7f3ef0cc2b90>
pre = CallNode(Op(equal), [Var(x, ty=TensorType([1, 128, 28, 28], float32)), Var(y, ty=TensorType([1, 128, 28, 28], float32))], (nullptr), [TensorType([1, 128, 28, 28], float32), TensorType([1, 128, 28, 28], float32)])
post = CallNode(Op(equal), [Var(x, ty=TensorType([1, 128, 28, 28], float32)), Var(y, ty=TensorType([1, 128, 28, 28], float32))], (nullptr), [TensorType([1, 128, 28, 28], float32), TensorType([1, 128, 28, 28], float32)])
node_map = {relay.dataflow_pattern.ExprPattern(0x55a8fca0fc40): [Op(equal)], relay.dataflow_pattern.WildcardPattern(0x55a8fca0fba... 128, 28, 28], float32))], (nullptr), [TensorType([1, 128, 28, 28], float32), TensorType([1, 128, 28, 28], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:3424: NameError

Check failure on line 232 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_equal[shape2] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape = (1, 256, 28, 28)

    @pytest.mark.parametrize(
        "shape",
        [
            (1, 128, 28, 28),
            (1, 64, 28, 28),
            (1, 256, 28, 28),
            (1, 128, 14, 14),
            (1, 128, 56, 56),
            (1, 32, 64, 64),
            (1, 512, 7, 7),
            (1, 32, 32, 32),
            (128, 28, 28),
            (64, 28, 28),
            (256, 28, 28),
            (128, 14, 14),
            (128, 56, 56),
            (32, 64, 64),
            (512, 7, 7),
            (32, 32, 32),
            (128, 28),
            (64, 28),
            (256, 28),
            (128, 14),
            (128, 56),
            (32, 64),
            (512, 7),
            (32, 32),
        ],
    )
    @pytest.mark.push
    def test_equal(shape):
        class Equal(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.eq(x, y)
    
        x = torch.rand(shape)
        y = x * 2.0
    
        inputs = [x, y]
    
        framework_model = Equal()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:232: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1095: in compile_for_forge
    compiled_relay_module = run_forge_compile_passes(relay_module, params, inputs, target, framework_outputs, verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4238: in run_forge_compile_passes
    return run_pattern_callbacks(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4229: in run_pattern_callbacks
    raise ex
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4226: in run_pattern_callbacks
    relay_module = _run_pattern_callback(relay_module, callback, callback_name)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4205: in _run_pattern_callback
    relay_module['main'] = rewrite(callback, relay_module['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge_passes.InverseMaskGen object at 0x7f3ef8a294b0>
pre = CallNode(Op(equal), [Var(x, ty=TensorType([1, 256, 28, 28], float32)), Var(y, ty=TensorType([1, 256, 28, 28], float32))], (nullptr), [TensorType([1, 256, 28, 28], float32), TensorType([1, 256, 28, 28], float32)])
post = CallNode(Op(equal), [Var(x, ty=TensorType([1, 256, 28, 28], float32)), Var(y, ty=TensorType([1, 256, 28, 28], float32))], (nullptr), [TensorType([1, 256, 28, 28], float32), TensorType([1, 256, 28, 28], float32)])
node_map = {relay.dataflow_pattern.ExprPattern(0x55a8fd2452a0): [Op(equal)], relay.dataflow_pattern.WildcardPattern(0x55a8fd24520... 256, 28, 28], float32))], (nullptr), [TensorType([1, 256, 28, 28], float32), TensorType([1, 256, 28, 28], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:3424: NameError

Check failure on line 232 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_equal[shape4] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape = (1, 128, 56, 56)

    @pytest.mark.parametrize(
        "shape",
        [
            (1, 128, 28, 28),
            (1, 64, 28, 28),
            (1, 256, 28, 28),
            (1, 128, 14, 14),
            (1, 128, 56, 56),
            (1, 32, 64, 64),
            (1, 512, 7, 7),
            (1, 32, 32, 32),
            (128, 28, 28),
            (64, 28, 28),
            (256, 28, 28),
            (128, 14, 14),
            (128, 56, 56),
            (32, 64, 64),
            (512, 7, 7),
            (32, 32, 32),
            (128, 28),
            (64, 28),
            (256, 28),
            (128, 14),
            (128, 56),
            (32, 64),
            (512, 7),
            (32, 32),
        ],
    )
    @pytest.mark.push
    def test_equal(shape):
        class Equal(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.eq(x, y)
    
        x = torch.rand(shape)
        y = x * 2.0
    
        inputs = [x, y]
    
        framework_model = Equal()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:232: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1095: in compile_for_forge
    compiled_relay_module = run_forge_compile_passes(relay_module, params, inputs, target, framework_outputs, verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4238: in run_forge_compile_passes
    return run_pattern_callbacks(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4229: in run_pattern_callbacks
    raise ex
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4226: in run_pattern_callbacks
    relay_module = _run_pattern_callback(relay_module, callback, callback_name)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4205: in _run_pattern_callback
    relay_module['main'] = rewrite(callback, relay_module['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge_passes.InverseMaskGen object at 0x7f3f1c0dc4c0>
pre = CallNode(Op(equal), [Var(x, ty=TensorType([1, 128, 56, 56], float32)), Var(y, ty=TensorType([1, 128, 56, 56], float32))], (nullptr), [TensorType([1, 128, 56, 56], float32), TensorType([1, 128, 56, 56], float32)])
post = CallNode(Op(equal), [Var(x, ty=TensorType([1, 128, 56, 56], float32)), Var(y, ty=TensorType([1, 128, 56, 56], float32))], (nullptr), [TensorType([1, 128, 56, 56], float32), TensorType([1, 128, 56, 56], float32)])
node_map = {relay.dataflow_pattern.WildcardPattern(0x55a8fdcf3380): [Var(y, ty=TensorType([1, 128, 56, 56], float32))], relay.dat...): [Var(x, ty=TensorType([1, 128, 56, 56], float32))], relay.dataflow_pattern.ExprPattern(0x55a8fdcf33a0): [Op(equal)]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:3424: NameError

Check failure on line 232 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_equal[shape6] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape = (1, 512, 7, 7)

    @pytest.mark.parametrize(
        "shape",
        [
            (1, 128, 28, 28),
            (1, 64, 28, 28),
            (1, 256, 28, 28),
            (1, 128, 14, 14),
            (1, 128, 56, 56),
            (1, 32, 64, 64),
            (1, 512, 7, 7),
            (1, 32, 32, 32),
            (128, 28, 28),
            (64, 28, 28),
            (256, 28, 28),
            (128, 14, 14),
            (128, 56, 56),
            (32, 64, 64),
            (512, 7, 7),
            (32, 32, 32),
            (128, 28),
            (64, 28),
            (256, 28),
            (128, 14),
            (128, 56),
            (32, 64),
            (512, 7),
            (32, 32),
        ],
    )
    @pytest.mark.push
    def test_equal(shape):
        class Equal(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.eq(x, y)
    
        x = torch.rand(shape)
        y = x * 2.0
    
        inputs = [x, y]
    
        framework_model = Equal()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:232: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1095: in compile_for_forge
    compiled_relay_module = run_forge_compile_passes(relay_module, params, inputs, target, framework_outputs, verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4238: in run_forge_compile_passes
    return run_pattern_callbacks(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4229: in run_pattern_callbacks
    raise ex
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4226: in run_pattern_callbacks
    relay_module = _run_pattern_callback(relay_module, callback, callback_name)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4205: in _run_pattern_callback
    relay_module['main'] = rewrite(callback, relay_module['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge_passes.InverseMaskGen object at 0x7f3ef02abee0>
pre = CallNode(Op(equal), [Var(x, ty=TensorType([1, 512, 7, 7], float32)), Var(y, ty=TensorType([1, 512, 7, 7], float32))], (nullptr), [TensorType([1, 512, 7, 7], float32), TensorType([1, 512, 7, 7], float32)])
post = CallNode(Op(equal), [Var(x, ty=TensorType([1, 512, 7, 7], float32)), Var(y, ty=TensorType([1, 512, 7, 7], float32))], (nullptr), [TensorType([1, 512, 7, 7], float32), TensorType([1, 512, 7, 7], float32)])
node_map = {relay.dataflow_pattern.WildcardPattern(0x55a8fca54da0): [Var(x, ty=TensorType([1, 512, 7, 7], float32))], relay.dataf...pe([1, 512, 7, 7], float32))], (nullptr), [TensorType([1, 512, 7, 7], float32), TensorType([1, 512, 7, 7], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:3424: NameError

Check failure on line 232 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_equal[shape8] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape = (128, 28, 28)

    @pytest.mark.parametrize(
        "shape",
        [
            (1, 128, 28, 28),
            (1, 64, 28, 28),
            (1, 256, 28, 28),
            (1, 128, 14, 14),
            (1, 128, 56, 56),
            (1, 32, 64, 64),
            (1, 512, 7, 7),
            (1, 32, 32, 32),
            (128, 28, 28),
            (64, 28, 28),
            (256, 28, 28),
            (128, 14, 14),
            (128, 56, 56),
            (32, 64, 64),
            (512, 7, 7),
            (32, 32, 32),
            (128, 28),
            (64, 28),
            (256, 28),
            (128, 14),
            (128, 56),
            (32, 64),
            (512, 7),
            (32, 32),
        ],
    )
    @pytest.mark.push
    def test_equal(shape):
        class Equal(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.eq(x, y)
    
        x = torch.rand(shape)
        y = x * 2.0
    
        inputs = [x, y]
    
        framework_model = Equal()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:232: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1095: in compile_for_forge
    compiled_relay_module = run_forge_compile_passes(relay_module, params, inputs, target, framework_outputs, verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4238: in run_forge_compile_passes
    return run_pattern_callbacks(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4229: in run_pattern_callbacks
    raise ex
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4226: in run_pattern_callbacks
    relay_module = _run_pattern_callback(relay_module, callback, callback_name)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4205: in _run_pattern_callback
    relay_module['main'] = rewrite(callback, relay_module['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge_passes.InverseMaskGen object at 0x7f3ef0bd4eb0>
pre = CallNode(Op(equal), [Var(x, ty=TensorType([128, 28, 28], float32)), Var(y, ty=TensorType([128, 28, 28], float32))], (nullptr), [TensorType([128, 28, 28], float32), TensorType([128, 28, 28], float32)])
post = CallNode(Op(equal), [Var(x, ty=TensorType([128, 28, 28], float32)), Var(y, ty=TensorType([128, 28, 28], float32))], (nullptr), [TensorType([128, 28, 28], float32), TensorType([128, 28, 28], float32)])
node_map = {relay.dataflow_pattern.WildcardPattern(0x55a8fcd13550): [Var(y, ty=TensorType([128, 28, 28], float32))], relay.datafl...rType([128, 28, 28], float32))], (nullptr), [TensorType([128, 28, 28], float32), TensorType([128, 28, 28], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:3424: NameError

Check failure on line 232 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_equal[shape11] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape = (128, 14, 14)

    @pytest.mark.parametrize(
        "shape",
        [
            (1, 128, 28, 28),
            (1, 64, 28, 28),
            (1, 256, 28, 28),
            (1, 128, 14, 14),
            (1, 128, 56, 56),
            (1, 32, 64, 64),
            (1, 512, 7, 7),
            (1, 32, 32, 32),
            (128, 28, 28),
            (64, 28, 28),
            (256, 28, 28),
            (128, 14, 14),
            (128, 56, 56),
            (32, 64, 64),
            (512, 7, 7),
            (32, 32, 32),
            (128, 28),
            (64, 28),
            (256, 28),
            (128, 14),
            (128, 56),
            (32, 64),
            (512, 7),
            (32, 32),
        ],
    )
    @pytest.mark.push
    def test_equal(shape):
        class Equal(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.eq(x, y)
    
        x = torch.rand(shape)
        y = x * 2.0
    
        inputs = [x, y]
    
        framework_model = Equal()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:232: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1095: in compile_for_forge
    compiled_relay_module = run_forge_compile_passes(relay_module, params, inputs, target, framework_outputs, verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4238: in run_forge_compile_passes
    return run_pattern_callbacks(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4229: in run_pattern_callbacks
    raise ex
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4226: in run_pattern_callbacks
    relay_module = _run_pattern_callback(relay_module, callback, callback_name)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4205: in _run_pattern_callback
    relay_module['main'] = rewrite(callback, relay_module['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge_passes.InverseMaskGen object at 0x7f3ef0f5c1f0>
pre = CallNode(Op(equal), [Var(x, ty=TensorType([128, 14, 14], float32)), Var(y, ty=TensorType([128, 14, 14], float32))], (nullptr), [TensorType([128, 14, 14], float32), TensorType([128, 14, 14], float32)])
post = CallNode(Op(equal), [Var(x, ty=TensorType([128, 14, 14], float32)), Var(y, ty=TensorType([128, 14, 14], float32))], (nullptr), [TensorType([128, 14, 14], float32), TensorType([128, 14, 14], float32)])
node_map = {relay.dataflow_pattern.WildcardPattern(0x55a8fd068250): [Var(x, ty=TensorType([128, 14, 14], float32))], relay.datafl...rType([128, 14, 14], float32))], (nullptr), [TensorType([128, 14, 14], float32), TensorType([128, 14, 14], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:3424: NameError

Check failure on line 232 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_equal[shape13] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape = (32, 64, 64)

    @pytest.mark.parametrize(
        "shape",
        [
            (1, 128, 28, 28),
            (1, 64, 28, 28),
            (1, 256, 28, 28),
            (1, 128, 14, 14),
            (1, 128, 56, 56),
            (1, 32, 64, 64),
            (1, 512, 7, 7),
            (1, 32, 32, 32),
            (128, 28, 28),
            (64, 28, 28),
            (256, 28, 28),
            (128, 14, 14),
            (128, 56, 56),
            (32, 64, 64),
            (512, 7, 7),
            (32, 32, 32),
            (128, 28),
            (64, 28),
            (256, 28),
            (128, 14),
            (128, 56),
            (32, 64),
            (512, 7),
            (32, 32),
        ],
    )
    @pytest.mark.push
    def test_equal(shape):
        class Equal(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.eq(x, y)
    
        x = torch.rand(shape)
        y = x * 2.0
    
        inputs = [x, y]
    
        framework_model = Equal()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:232: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1095: in compile_for_forge
    compiled_relay_module = run_forge_compile_passes(relay_module, params, inputs, target, framework_outputs, verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4238: in run_forge_compile_passes
    return run_pattern_callbacks(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4229: in run_pattern_callbacks
    raise ex
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4226: in run_pattern_callbacks
    relay_module = _run_pattern_callback(relay_module, callback, callback_name)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4205: in _run_pattern_callback
    relay_module['main'] = rewrite(callback, relay_module['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge_passes.InverseMaskGen object at 0x7f3ef02afca0>
pre = CallNode(Op(equal), [Var(x, ty=TensorType([32, 64, 64], float32)), Var(y, ty=TensorType([32, 64, 64], float32))], (nullptr), [TensorType([32, 64, 64], float32), TensorType([32, 64, 64], float32)])
post = CallNode(Op(equal), [Var(x, ty=TensorType([32, 64, 64], float32)), Var(y, ty=TensorType([32, 64, 64], float32))], (nullptr), [TensorType([32, 64, 64], float32), TensorType([32, 64, 64], float32)])
node_map = {relay.dataflow_pattern.CallPattern(0x55a8ffb90330): [CallNode(Op(equal), [Var(x, ty=TensorType([32, 64, 64], float32)...4], float32))], relay.dataflow_pattern.WildcardPattern(0x55a8fdce4710): [Var(x, ty=TensorType([32, 64, 64], float32))]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:3424: NameError

Check failure on line 232 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_equal[shape15] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape = (32, 32, 32)

    @pytest.mark.parametrize(
        "shape",
        [
            (1, 128, 28, 28),
            (1, 64, 28, 28),
            (1, 256, 28, 28),
            (1, 128, 14, 14),
            (1, 128, 56, 56),
            (1, 32, 64, 64),
            (1, 512, 7, 7),
            (1, 32, 32, 32),
            (128, 28, 28),
            (64, 28, 28),
            (256, 28, 28),
            (128, 14, 14),
            (128, 56, 56),
            (32, 64, 64),
            (512, 7, 7),
            (32, 32, 32),
            (128, 28),
            (64, 28),
            (256, 28),
            (128, 14),
            (128, 56),
            (32, 64),
            (512, 7),
            (32, 32),
        ],
    )
    @pytest.mark.push
    def test_equal(shape):
        class Equal(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.eq(x, y)
    
        x = torch.rand(shape)
        y = x * 2.0
    
        inputs = [x, y]
    
        framework_model = Equal()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:232: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1095: in compile_for_forge
    compiled_relay_module = run_forge_compile_passes(relay_module, params, inputs, target, framework_outputs, verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4238: in run_forge_compile_passes
    return run_pattern_callbacks(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4229: in run_pattern_callbacks
    raise ex
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4226: in run_pattern_callbacks
    relay_module = _run_pattern_callback(relay_module, callback, callback_name)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4205: in _run_pattern_callback
    relay_module['main'] = rewrite(callback, relay_module['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge_passes.InverseMaskGen object at 0x7f3ef0b4c160>
pre = CallNode(Op(equal), [Var(x, ty=TensorType([32, 32, 32], float32)), Var(y, ty=TensorType([32, 32, 32], float32))], (nullptr), [TensorType([32, 32, 32], float32), TensorType([32, 32, 32], float32)])
post = CallNode(Op(equal), [Var(x, ty=TensorType([32, 32, 32], float32)), Var(y, ty=TensorType([32, 32, 32], float32))], (nullptr), [TensorType([32, 32, 32], float32), TensorType([32, 32, 32], float32)])
node_map = {relay.dataflow_pattern.WildcardPattern(0x55a8fcf65cc0): [Var(y, ty=TensorType([32, 32, 32], float32))], relay.dataflo..., 32], float32), TensorType([32, 32, 32], float32)])], relay.dataflow_pattern.ExprPattern(0x55a8fd06d320): [Op(equal)]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:3424: NameError

Check failure on line 232 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_equal[shape17] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape = (64, 28)

    @pytest.mark.parametrize(
        "shape",
        [
            (1, 128, 28, 28),
            (1, 64, 28, 28),
            (1, 256, 28, 28),
            (1, 128, 14, 14),
            (1, 128, 56, 56),
            (1, 32, 64, 64),
            (1, 512, 7, 7),
            (1, 32, 32, 32),
            (128, 28, 28),
            (64, 28, 28),
            (256, 28, 28),
            (128, 14, 14),
            (128, 56, 56),
            (32, 64, 64),
            (512, 7, 7),
            (32, 32, 32),
            (128, 28),
            (64, 28),
            (256, 28),
            (128, 14),
            (128, 56),
            (32, 64),
            (512, 7),
            (32, 32),
        ],
    )
    @pytest.mark.push
    def test_equal(shape):
        class Equal(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.eq(x, y)
    
        x = torch.rand(shape)
        y = x * 2.0
    
        inputs = [x, y]
    
        framework_model = Equal()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:232: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1095: in compile_for_forge
    compiled_relay_module = run_forge_compile_passes(relay_module, params, inputs, target, framework_outputs, verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4238: in run_forge_compile_passes
    return run_pattern_callbacks(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4229: in run_pattern_callbacks
    raise ex
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4226: in run_pattern_callbacks
    relay_module = _run_pattern_callback(relay_module, callback, callback_name)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4205: in _run_pattern_callback
    relay_module['main'] = rewrite(callback, relay_module['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge_passes.InverseMaskGen object at 0x7f3f1c0ebdf0>
pre = CallNode(Op(equal), [Var(x, ty=TensorType([64, 28], float32)), Var(y, ty=TensorType([64, 28], float32))], (nullptr), [TensorType([64, 28], float32), TensorType([64, 28], float32)])
post = CallNode(Op(equal), [Var(x, ty=TensorType([64, 28], float32)), Var(y, ty=TensorType([64, 28], float32))], (nullptr), [TensorType([64, 28], float32), TensorType([64, 28], float32)])
node_map = {relay.dataflow_pattern.ExprPattern(0x55a8fdcf2830): [Op(equal)], relay.dataflow_pattern.WildcardPattern(0x55a8fdcf281...Var(y, ty=TensorType([64, 28], float32))], (nullptr), [TensorType([64, 28], float32), TensorType([64, 28], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:3424: NameError

Check failure on line 232 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_equal[shape19] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape = (128, 14)

    @pytest.mark.parametrize(
        "shape",
        [
            (1, 128, 28, 28),
            (1, 64, 28, 28),
            (1, 256, 28, 28),
            (1, 128, 14, 14),
            (1, 128, 56, 56),
            (1, 32, 64, 64),
            (1, 512, 7, 7),
            (1, 32, 32, 32),
            (128, 28, 28),
            (64, 28, 28),
            (256, 28, 28),
            (128, 14, 14),
            (128, 56, 56),
            (32, 64, 64),
            (512, 7, 7),
            (32, 32, 32),
            (128, 28),
            (64, 28),
            (256, 28),
            (128, 14),
            (128, 56),
            (32, 64),
            (512, 7),
            (32, 32),
        ],
    )
    @pytest.mark.push
    def test_equal(shape):
        class Equal(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.eq(x, y)
    
        x = torch.rand(shape)
        y = x * 2.0
    
        inputs = [x, y]
    
        framework_model = Equal()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:232: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1095: in compile_for_forge
    compiled_relay_module = run_forge_compile_passes(relay_module, params, inputs, target, framework_outputs, verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4238: in run_forge_compile_passes
    return run_pattern_callbacks(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4229: in run_pattern_callbacks
    raise ex
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4226: in run_pattern_callbacks
    relay_module = _run_pattern_callback(relay_module, callback, callback_name)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4205: in _run_pattern_callback
    relay_module['main'] = rewrite(callback, relay_module['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge_passes.InverseMaskGen object at 0x7f3ef0e19f60>
pre = CallNode(Op(equal), [Var(x, ty=TensorType([128, 14], float32)), Var(y, ty=TensorType([128, 14], float32))], (nullptr), [TensorType([128, 14], float32), TensorType([128, 14], float32)])
post = CallNode(Op(equal), [Var(x, ty=TensorType([128, 14], float32)), Var(y, ty=TensorType([128, 14], float32))], (nullptr), [TensorType([128, 14], float32), TensorType([128, 14], float32)])
node_map = {relay.dataflow_pattern.CallPattern(0x55a8fe0e5580): [CallNode(Op(equal), [Var(x, ty=TensorType([128, 14], float32)), ...560): [Op(equal)], relay.dataflow_pattern.WildcardPattern(0x55a8fd11d7d0): [Var(x, ty=TensorType([128, 14], float32))]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:3424: NameError

Check failure on line 232 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_equal[shape20] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape = (128, 56)

    @pytest.mark.parametrize(
        "shape",
        [
            (1, 128, 28, 28),
            (1, 64, 28, 28),
            (1, 256, 28, 28),
            (1, 128, 14, 14),
            (1, 128, 56, 56),
            (1, 32, 64, 64),
            (1, 512, 7, 7),
            (1, 32, 32, 32),
            (128, 28, 28),
            (64, 28, 28),
            (256, 28, 28),
            (128, 14, 14),
            (128, 56, 56),
            (32, 64, 64),
            (512, 7, 7),
            (32, 32, 32),
            (128, 28),
            (64, 28),
            (256, 28),
            (128, 14),
            (128, 56),
            (32, 64),
            (512, 7),
            (32, 32),
        ],
    )
    @pytest.mark.push
    def test_equal(shape):
        class Equal(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.eq(x, y)
    
        x = torch.rand(shape)
        y = x * 2.0
    
        inputs = [x, y]
    
        framework_model = Equal()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:232: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1095: in compile_for_forge
    compiled_relay_module = run_forge_compile_passes(relay_module, params, inputs, target, framework_outputs, verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4238: in run_forge_compile_passes
    return run_pattern_callbacks(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4229: in run_pattern_callbacks
    raise ex
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4226: in run_pattern_callbacks
    relay_module = _run_pattern_callback(relay_module, callback, callback_name)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4205: in _run_pattern_callback
    relay_module['main'] = rewrite(callback, relay_module['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge_passes.InverseMaskGen object at 0x7f3ef89a0550>
pre = CallNode(Op(equal), [Var(x, ty=TensorType([128, 56], float32)), Var(y, ty=TensorType([128, 56], float32))], (nullptr), [TensorType([128, 56], float32), TensorType([128, 56], float32)])
post = CallNode(Op(equal), [Var(x, ty=TensorType([128, 56], float32)), Var(y, ty=TensorType([128, 56], float32))], (nullptr), [TensorType([128, 56], float32), TensorType([128, 56], float32)])
node_map = {relay.dataflow_pattern.ExprPattern(0x55a8fdafd500): [Op(equal)], relay.dataflow_pattern.WildcardPattern(0x55a8fd4edb9...(y, ty=TensorType([128, 56], float32))], (nullptr), [TensorType([128, 56], float32), TensorType([128, 56], float32)])]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:3424: NameError

Check failure on line 232 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_equal[shape22] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
shape = (512, 7)

    @pytest.mark.parametrize(
        "shape",
        [
            (1, 128, 28, 28),
            (1, 64, 28, 28),
            (1, 256, 28, 28),
            (1, 128, 14, 14),
            (1, 128, 56, 56),
            (1, 32, 64, 64),
            (1, 512, 7, 7),
            (1, 32, 32, 32),
            (128, 28, 28),
            (64, 28, 28),
            (256, 28, 28),
            (128, 14, 14),
            (128, 56, 56),
            (32, 64, 64),
            (512, 7, 7),
            (32, 32, 32),
            (128, 28),
            (64, 28),
            (256, 28),
            (128, 14),
            (128, 56),
            (32, 64),
            (512, 7),
            (32, 32),
        ],
    )
    @pytest.mark.push
    def test_equal(shape):
        class Equal(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, x, y):
                return torch.eq(x, y)
    
        x = torch.rand(shape)
        y = x * 2.0
    
        inputs = [x, y]
    
        framework_model = Equal()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:232: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1095: in compile_for_forge
    compiled_relay_module = run_forge_compile_passes(relay_module, params, inputs, target, framework_outputs, verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4238: in run_forge_compile_passes
    return run_pattern_callbacks(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4229: in run_pattern_callbacks
    raise ex
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4226: in run_pattern_callbacks
    relay_module = _run_pattern_callback(relay_module, callback, callback_name)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:4205: in _run_pattern_callback
    relay_module['main'] = rewrite(callback, relay_module['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge_passes.InverseMaskGen object at 0x7f3ef891c790>
pre = CallNode(Op(equal), [Var(x, ty=TensorType([512, 7], float32)), Var(y, ty=TensorType([512, 7], float32))], (nullptr), [TensorType([512, 7], float32), TensorType([512, 7], float32)])
post = CallNode(Op(equal), [Var(x, ty=TensorType([512, 7], float32)), Var(y, ty=TensorType([512, 7], float32))], (nullptr), [TensorType([512, 7], float32), TensorType([512, 7], float32)])
node_map = {relay.dataflow_pattern.WildcardPattern(0x55a8fe0e7a30): [Var(x, ty=TensorType([512, 7], float32))], relay.dataflow_pa...3110): [Op(equal)], relay.dataflow_pattern.WildcardPattern(0x55a8fe0e7a50): [Var(y, ty=TensorType([512, 7], float32))]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge_passes.py:3424: NameError

Check failure on line 267 in forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py

See this annotation in the file changed.

@github-actions github-actions / TT-Forge-FE Tests

test_eltwise_binary.test_greater_equal[dims1] 

NameError: name 'construct_pre_node_map' is not defined
Raw output 
dims = (6, 33)

    @pytest.mark.parametrize("dims", [(1, 32, 64), (6, 33), (4, 16, 17)])
    @pytest.mark.push
    def test_greater_equal(dims):
        class GreaterEqual(nn.Module):
            def __init__(self):
                super().__init__()
    
            def forward(self, a, b):
                return torch.greater_equal(a, b)
    
        inputs = [torch.rand(dims), torch.rand(dims)]
    
        framework_model = GreaterEqual()
>       compiled_model = forge.compile(framework_model, sample_inputs=inputs)

forge/test/mlir/operators/eltwise_binary/test_eltwise_binary.py:267: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:251: in compile_main
    return forge_compile_from_context(compile_context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:293: in forge_compile_from_context
    next_stage = stage_to_func[current_stage](context)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:671: in generate_initial_graph
    module, module_inputs = convert_to_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/compile.py:1032: in convert_to_forge_module
    forge_module, dev_types, module_inputs = generate_forge_module(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2062: in generate_forge_module
    module_writers, flattened_inputs = compile_tvm_to_python(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_to_python.py:2151: in compile_tvm_to_python
    json_graphs, flattened_pytorch_inputs, weights = load_tvm_graph(
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:104: in load_tvm_graph
    json_graphs, flattened_inputs = compile_tvm_graph(inputs, module, compiler_cfg, graph_name=graph_name, input_names=input_names, path=path, verify_cfg=verify_cfg, framework=framework)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:144: in compile_tvm_graph
    json_graphs, inputs = compile_pytorch_for_forge(module, *inputs, graph_name=graph_name, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg, input_names=input_names)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:394: in compile_pytorch_for_forge
    partitioned_mod, forge_params = compile_tvm_for_forge(mod, params, np_inputs, framework_outputs, input_names=flattened_input_names, graph_name=graph_name, return_params=True, compiler_cfg=compiler_cfg, verify_cfg=verify_cfg)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/forge_compile.py:408: in compile_tvm_for_forge
    mod, params = compile_for_forge(mod, target=target, params=params, graph_name=graph_name, **verify_args)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1100: in compile_for_forge
    warn_of_int_comparisons(compiled_relay_module)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1068: in warn_of_int_comparisons
    rewrite(warner, mod['main'])
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/relay/dataflow_pattern/__init__.py:927: in rewrite
    return ffi.rewrite(tmp, expr, mod)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/_ctypes/packed_func.py:242: in __call__
    raise_last_ffi_error()
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/tvm/_ffi/base.py:479: in raise_last_ffi_error
    raise py_err
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <forge.tvm_calls.relay.op.forge.CompareWarner object at 0x7f3ef02dea70>
pre = CallNode(Op(greater_equal), [Var(a, ty=TensorType([6, 33], float32)), Var(b, ty=TensorType([6, 33], float32))], (nullptr), [TensorType([6, 33], float32), TensorType([6, 33], float32)])
post = CallNode(Op(greater_equal), [Var(a, ty=TensorType([6, 33], float32)), Var(b, ty=TensorType([6, 33], float32))], (nullptr), [TensorType([6, 33], float32), TensorType([6, 33], float32)])
node_map = {relay.dataflow_pattern.CallPattern(0x55a8f9e00e20): [CallNode(Op(greater_equal), [Var(a, ty=TensorType([6, 33], float...0): [Var(b, ty=TensorType([6, 33], float32))], relay.dataflow_pattern.ExprPattern(0x55a8fcb5f7f0): [Op(greater_equal)]}

    def callback(self, pre, post, node_map):
>       pre_node_map = construct_pre_node_map(self.pattern, pre)
E       NameError: name 'construct_pre_node_map' is not defined

/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/forge/tvm_calls/relay/op/forge.py:1058: NameError
View more details on GitHub Actions