X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;ds=sidebyside;f=drivers%2Fiommu%2Famd_iommu_v2.c;h=1131664b918b0a574c7cc654a6a3cd04107f8e81;hb=bfa7df01a092e92ae764d39fee07fc3f486ed6a9;hp=f7b875bb70d42138027f49ebde8150d27ce14cd2;hpb=0c8e2f2c7b86c29775379f716f7db07516fcbcbf;p=firefly-linux-kernel-4.4.55.git

diff --git a/drivers/iommu/amd_iommu_v2.c b/drivers/iommu/amd_iommu_v2.c
index f7b875bb70d4..1131664b918b 100644
--- a/drivers/iommu/amd_iommu_v2.c
+++ b/drivers/iommu/amd_iommu_v2.c
@@ -356,8 +356,8 @@ static void free_pasid_states(struct device_state *dev_state)
 		free_pasid_states_level2(dev_state->states);
 	else if (dev_state->pasid_levels == 1)
 		free_pasid_states_level1(dev_state->states);
-	else if (dev_state->pasid_levels != 0)
-		BUG();
+	else
+		BUG_ON(dev_state->pasid_levels != 0);
 
 	free_page((unsigned long)dev_state->states);
 }