Skip to content

Commit 79bfea2

Browse files
fabtests/common: enable rocr device interface for fabtests
Signed-off-by: alexander-sannikov <[email protected]>
1 parent 7d60749 commit 79bfea2

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

fabtests/common/shared.c

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3414,7 +3414,7 @@ void ft_usage(char *name, char *desc)
34143414
void ft_hmem_usage()
34153415
{
34163416
FT_PRINT_OPTS_USAGE("-D <device_iface>", "Specify device interface: "
3417-
"e.g. cuda, ze, neuron, synapseai (default: None). "
3417+
"e.g. cuda, ze, neuron, synapseai, rocr (default: None). "
34183418
"Automatically enables FI_HMEM (-H)");
34193419
FT_PRINT_OPTS_USAGE("-i <device_id>", "Specify which device to use (default: 0)");
34203420
FT_PRINT_OPTS_USAGE("-H", "Enable provider FI_HMEM support");
@@ -3576,6 +3576,8 @@ void ft_parse_hmem_opts(int op, char *optarg, struct ft_opts *opts)
35763576
opts->iface = FI_HMEM_CUDA;
35773577
else if (!strncasecmp("neuron", optarg, 6))
35783578
opts->iface = FI_HMEM_NEURON;
3579+
else if (!strncasecmp("rocr", optarg, 4))
3580+
opts->iface = FI_HMEM_ROCR;
35793581
else if (!strncasecmp("synapseai", optarg, 9)) {
35803582
opts->iface = FI_HMEM_SYNAPSEAI;
35813583
opts->options |= FT_OPT_REG_DMABUF_MR;

0 commit comments

Comments
 (0)