47 	struct pci_dev *gpdev;  in pnv_pci_get_gpu_dev()  local
60 	gpdev = get_pci_dev(dn);  in pnv_pci_get_gpu_dev()
63 	return gpdev;  in pnv_pci_get_gpu_dev()
68 struct pci_dev *pnv_pci_get_npu_dev(struct pci_dev *gpdev, int index)  in pnv_pci_get_npu_dev()  argument
73 	if (WARN_ON(!gpdev))  in pnv_pci_get_npu_dev()
77 	if (!gpdev->dev.of_node)  in pnv_pci_get_npu_dev()
81 	dn = of_parse_phandle(gpdev->dev.of_node, "ibm,npu", index);  in pnv_pci_get_npu_dev()
98 						  struct pci_dev **gpdev)  in get_gpu_pci_dev_and_pe()  argument
118 	if (gpdev)  in get_gpu_pci_dev_and_pe()
119 		*gpdev = pdev;  in get_gpu_pci_dev_and_pe()
203 	struct pci_dev *gpdev = NULL;  in pnv_npu_take_ownership()  local
226 	get_gpu_pci_dev_and_pe(npe, &gpdev);  in pnv_npu_take_ownership()
227 	if (gpdev)  in pnv_npu_take_ownership()
228 		pnv_npu2_unmap_lpar_dev(gpdev);  in pnv_npu_take_ownership()
235 	struct pci_dev *gpdev = NULL;  in pnv_npu_release_ownership()  local
237 	get_gpu_pci_dev_and_pe(npe, &gpdev);  in pnv_npu_release_ownership()
238 	if (gpdev)  in pnv_npu_release_ownership()
239 		pnv_npu2_map_lpar_dev(gpdev, 0, MSR_DR | MSR_PR | MSR_HV);  in pnv_npu_release_ownership()
432 	struct pci_dev *gpdev = NULL;  in pnv_try_setup_npu_table_group()  local
436 	list_for_each_entry(gpdev, &pe->pbus->devices, bus_list) {  in pnv_try_setup_npu_table_group()
437 		npdev = pnv_pci_get_npu_dev(gpdev, 0);  in pnv_try_setup_npu_table_group()
485 	struct pci_dev *gpdev = NULL;  in pnv_npu_compound_attach()  local
487 	struct pnv_ioda_pe *gpe = get_gpu_pci_dev_and_pe(pe, &gpdev);  in pnv_npu_compound_attach()
502 			iommu_group_get(&gpdev->dev));  in pnv_npu_compound_attach()
516 		if (gpdevtmp != gpdev)  in pnv_npu_compound_attach()
551 int pnv_npu2_map_lpar_dev(struct pci_dev *gpdev, unsigned int lparid,  in pnv_npu2_map_lpar_dev()  argument
555 	struct pci_dev *npdev = pnv_pci_get_npu_dev(gpdev, 0);  in pnv_npu2_map_lpar_dev()
565 	dev_dbg(&gpdev->dev, "Map LPAR opalid=%llu lparid=%u\n",  in pnv_npu2_map_lpar_dev()
571 	ret = opal_npu_map_lpar(nphb->opal_id, pci_dev_id(gpdev), lparid,  in pnv_npu2_map_lpar_dev()
574 		dev_err(&gpdev->dev, "Error %d mapping device to LPAR\n", ret);  in pnv_npu2_map_lpar_dev()
578 	dev_dbg(&gpdev->dev, "init context opalid=%llu msr=%lx\n",  in pnv_npu2_map_lpar_dev()
581 				    pci_dev_id(gpdev));  in pnv_npu2_map_lpar_dev()
583 		dev_err(&gpdev->dev, "Failed to init context: %d\n", ret);  in pnv_npu2_map_lpar_dev()
593 	struct pci_dev *gpdev;  in pnv_npu2_map_lpar()  local
595 	list_for_each_entry(gpdev, &gpe->pbus->devices, bus_list)  in pnv_npu2_map_lpar()
596 		pnv_npu2_map_lpar_dev(gpdev, 0, msr);  in pnv_npu2_map_lpar()
599 int pnv_npu2_unmap_lpar_dev(struct pci_dev *gpdev)  in pnv_npu2_unmap_lpar_dev()  argument
602 	struct pci_dev *npdev = pnv_pci_get_npu_dev(gpdev, 0);  in pnv_npu2_unmap_lpar_dev()
612 	dev_dbg(&gpdev->dev, "destroy context opalid=%llu\n",  in pnv_npu2_unmap_lpar_dev()
615 				       pci_dev_id(gpdev));  in pnv_npu2_unmap_lpar_dev()
617 		dev_err(&gpdev->dev, "Failed to destroy context: %d\n", ret);  in pnv_npu2_unmap_lpar_dev()
622 	dev_dbg(&gpdev->dev, "Map LPAR opalid=%llu lparid=0\n", nphb->opal_id);  in pnv_npu2_unmap_lpar_dev()
623 	ret = opal_npu_map_lpar(nphb->opal_id, pci_dev_id(gpdev), 0 /*LPID*/,  in pnv_npu2_unmap_lpar_dev()
626 		dev_err(&gpdev->dev, "Error %d mapping device to LPAR\n", ret);  in pnv_npu2_unmap_lpar_dev()