02ae456ef0e4eef86455b0a39d5ccabfd5b29668,tests/python/gpu/test_fusion.py,,check_fused_symbol,#Any#,29

Before Change

        data = {inp : kwargs[inp].astype(dtype) for inp in inputs}
        for grad_req in ["write", "add"]:
            type_dict = {inp : dtype for inp in inputs}
            os.environ["MXNET_USE_FUSION"] = "0"
            orig_exec = test_sym._simple_bind(ctx=ctx, grad_req=grad_req, type_dict=type_dict, **shapes)
            os.environ["MXNET_USE_FUSION"] = "1"
            fused_exec = test_sym._simple_bind(ctx=ctx, grad_req=grad_req, type_dict=type_dict, **shapes)
            fwd_orig = orig_exec.forward(is_train=True, **data)

After Change

            type_dict = {inp : dtype for inp in inputs}
            with environment("MXNET_USE_FUSION", "0"):
                orig_exec = test_sym._simple_bind(ctx=ctx, grad_req=grad_req, type_dict=type_dict, **shapes)
            with environment("MXNET_USE_FUSION", "1"):
                fused_exec = test_sym._simple_bind(ctx=ctx, grad_req=grad_req, type_dict=type_dict, **shapes)
            fwd_orig = orig_exec.forward(is_train=True, **data)
            out_grads = [mx.nd.ones_like(arr) for arr in fwd_orig]
            orig_exec.backward(out_grads=out_grads)
            fwd_fused = fused_exec.forward(is_train=True, **data)

Instances