48 	struct pci_dev *gpdev;  in pnv_pci_get_gpu_dev()  local
61 	gpdev = get_pci_dev(dn);  in pnv_pci_get_gpu_dev()
64 	return gpdev;  in pnv_pci_get_gpu_dev()
69 struct pci_dev *pnv_pci_get_npu_dev(struct pci_dev *gpdev, int index)  in pnv_pci_get_npu_dev()  argument
74 	if (WARN_ON(!gpdev))  in pnv_pci_get_npu_dev()
78 	if (!gpdev->dev.of_node)  in pnv_pci_get_npu_dev()
82 	dn = of_parse_phandle(gpdev->dev.of_node, "ibm,npu", index);  in pnv_pci_get_npu_dev()
99 						  struct pci_dev **gpdev)  in get_gpu_pci_dev_and_pe()  argument
119 	if (gpdev)  in get_gpu_pci_dev_and_pe()
120 		*gpdev = pdev;  in get_gpu_pci_dev_and_pe()
204 	struct pci_dev *gpdev = NULL;  in pnv_npu_take_ownership()  local
227 	get_gpu_pci_dev_and_pe(npe, &gpdev);  in pnv_npu_take_ownership()
228 	if (gpdev)  in pnv_npu_take_ownership()
229 		pnv_npu2_unmap_lpar_dev(gpdev);  in pnv_npu_take_ownership()
236 	struct pci_dev *gpdev = NULL;  in pnv_npu_release_ownership()  local
238 	get_gpu_pci_dev_and_pe(npe, &gpdev);  in pnv_npu_release_ownership()
239 	if (gpdev)  in pnv_npu_release_ownership()
240 		pnv_npu2_map_lpar_dev(gpdev, 0, MSR_DR | MSR_PR | MSR_HV);  in pnv_npu_release_ownership()
436 	struct pci_dev *gpdev = NULL;  in pnv_try_setup_npu_table_group()  local
440 	list_for_each_entry(gpdev, &pe->pbus->devices, bus_list) {  in pnv_try_setup_npu_table_group()
441 		npdev = pnv_pci_get_npu_dev(gpdev, 0);  in pnv_try_setup_npu_table_group()
480 	iommu_del_device(&gpdev->dev);  in pnv_try_setup_npu_table_group()
493 	iommu_add_device(compound_group, &gpdev->dev);  in pnv_try_setup_npu_table_group()
502 	struct pci_dev *gpdev = NULL;  in pnv_npu_compound_attach()  local
504 	struct pnv_ioda_pe *gpe = get_gpu_pci_dev_and_pe(pe, &gpdev);  in pnv_npu_compound_attach()
519 			iommu_group_get(&gpdev->dev));  in pnv_npu_compound_attach()
533 		if (gpdevtmp != gpdev)  in pnv_npu_compound_attach()
616 int pnv_npu2_map_lpar_dev(struct pci_dev *gpdev, unsigned int lparid,  in pnv_npu2_map_lpar_dev()  argument
620 	struct pci_dev *npdev = pnv_pci_get_npu_dev(gpdev, 0);  in pnv_npu2_map_lpar_dev()
635 	dev_dbg(&gpdev->dev, "Map LPAR opalid=%llu lparid=%u\n",  in pnv_npu2_map_lpar_dev()
641 	ret = opal_npu_map_lpar(nphb->opal_id, pci_dev_id(gpdev), lparid,  in pnv_npu2_map_lpar_dev()
644 		dev_err(&gpdev->dev, "Error %d mapping device to LPAR\n", ret);  in pnv_npu2_map_lpar_dev()
648 	dev_dbg(&gpdev->dev, "init context opalid=%llu msr=%lx\n",  in pnv_npu2_map_lpar_dev()
651 				    pci_dev_id(gpdev));  in pnv_npu2_map_lpar_dev()
653 		dev_err(&gpdev->dev, "Failed to init context: %d\n", ret);  in pnv_npu2_map_lpar_dev()
663 	struct pci_dev *gpdev;  in pnv_npu2_map_lpar()  local
665 	list_for_each_entry(gpdev, &gpe->pbus->devices, bus_list)  in pnv_npu2_map_lpar()
666 		pnv_npu2_map_lpar_dev(gpdev, 0, msr);  in pnv_npu2_map_lpar()
669 int pnv_npu2_unmap_lpar_dev(struct pci_dev *gpdev)  in pnv_npu2_unmap_lpar_dev()  argument
672 	struct pci_dev *npdev = pnv_pci_get_npu_dev(gpdev, 0);  in pnv_npu2_unmap_lpar_dev()
687 	dev_dbg(&gpdev->dev, "destroy context opalid=%llu\n",  in pnv_npu2_unmap_lpar_dev()
690 				       pci_dev_id(gpdev));  in pnv_npu2_unmap_lpar_dev()
692 		dev_err(&gpdev->dev, "Failed to destroy context: %d\n", ret);  in pnv_npu2_unmap_lpar_dev()
697 	dev_dbg(&gpdev->dev, "Map LPAR opalid=%llu lparid=0\n", nphb->opal_id);  in pnv_npu2_unmap_lpar_dev()
698 	ret = opal_npu_map_lpar(nphb->opal_id, pci_dev_id(gpdev), 0 /*LPID*/,  in pnv_npu2_unmap_lpar_dev()
701 		dev_err(&gpdev->dev, "Error %d mapping device to LPAR\n", ret);  in pnv_npu2_unmap_lpar_dev()