Skip to content

Commit 919a4f5

Browse files
committed
Merge branch 'dev-nvbug-5181418' into 'main'
[NVBUG 5181418] Fix the *forcepci* unit test failures See merge request gpudirect/gdrcopy!60
2 parents 0fd49d5 + debdf8a commit 919a4f5

File tree

1 file changed

+12
-5
lines changed

1 file changed

+12
-5
lines changed

src/gdrdrv/gdrdrv.c

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -460,6 +460,17 @@ static inline int gdr_use_persistent_mapping(void)
460460

461461
//-----------------------------------------------------------------------------
462462

463+
static inline int gdr_support_force_pcie(void)
464+
{
465+
#ifdef NVIDIA_P2P_FLAGS_FORCE_BAR1_MAPPING
466+
return gdr_use_persistent_mapping();
467+
#else
468+
return 0;
469+
#endif
470+
}
471+
472+
//-----------------------------------------------------------------------------
473+
463474
static int gdrdrv_open(struct inode *inode, struct file *filp)
464475
{
465476
unsigned int minor = MINOR(inode->i_rdev);
@@ -1201,11 +1212,7 @@ static int gdrdrv_get_attr(gdr_info_t *info, void __user *_params)
12011212
params.val = gdr_use_persistent_mapping();
12021213
break;
12031214
case GDRDRV_ATTR_SUPPORT_PIN_FLAG_FORCE_PCIE:
1204-
#ifdef NVIDIA_P2P_FLAGS_FORCE_BAR1_MAPPING
1205-
params.val = 1;
1206-
#else
1207-
params.val = 0;
1208-
#endif
1215+
params.val = gdr_support_force_pcie();
12091216
break;
12101217
default:
12111218
ret = -EINVAL;

0 commit comments

Comments
 (0)