5e0b54bdc95ebd072fc4f3698b1e1bca5cb3adcd,tests/chainermn_tests/links_tests/test_create_mnbn_model.py,TestCreateMnBnModel,check_create_mnbn_model_sequential,#TestCreateMnBnModel#Any#,76

Before Change

        mnbn_model = chainermn.links.create_mnbn_model(model,
                                                       self.communicator)

        if gpu:
            device_id = self.communicator.intra_rank
            mnbn_model.to_gpu(device=device_id)
        else:
            device_id = -1
        with chainer.using_device(mnbn_model.device):
            x = mnbn_model.xp.zeros((1, 1, 1, 1))
            mnbn_model(x)

After Change

        mnbn_model = chainermn.links.create_mnbn_model(model,
                                                       self.communicator)

        if gpu:
            mnbn_model.to_device(cupy.cuda.Device())
        with chainer.using_device(mnbn_model.device):
            x = mnbn_model.xp.zeros((1, 1, 1, 1))
            mnbn_model(x)

Instances