Update wrapper fi_close

HabanaAI · Mar 23, 2023 · dddc1e9 · dddc1e9
1 parent d7798f1
commit dddc1e9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/hccl_ofi_wrapper.cpp b/hccl_ofi_wrapper.cpp
@@ -37,9 +37,9 @@ char* ofi_plugin::w_fi_tostr(const void* data, enum fi_type datatype)
     return fi_tostr(data, datatype);
 }
 
-void ofi_plugin::w_fi_close(fid_t domain)
+int ofi_plugin::w_fi_close(fid_t domain)
 {
-    fi_close(domain);
+    return fi_close(domain);
 }
 
 int ofi_plugin::w_fi_fabric(struct fi_fabric_attr* attr, struct fid_fabric** fabric, void* context)

diff --git a/hccl_ofi_wrapper.h b/hccl_ofi_wrapper.h
@@ -28,7 +28,7 @@ class ofi_plugin : public ofi_plugin_interface
     void            w_fi_freeinfo(struct fi_info* info);
     const char*     w_fi_strerror(int err);
     char*           w_fi_tostr(const void* data, enum fi_type datatype);
-    void            w_fi_close(fid_t domain);
+    int             w_fi_close(fid_t domain);
     int             w_fi_fabric(struct fi_fabric_attr* attr, struct fid_fabric** fabric, void* context);
 
     int w_fi_domain(struct fid_fabric* fabric, struct fi_info* info, struct fid_domain** domain, void* context);

diff --git a/hccl_ofi_wrapper_interface.h b/hccl_ofi_wrapper_interface.h
@@ -34,7 +34,7 @@ class ofi_plugin_interface
     virtual void            w_fi_freeinfo(struct fi_info* info)                                                 = 0;
     virtual const char*     w_fi_strerror(int err)                                                              = 0;
     virtual char*           w_fi_tostr(const void* data, enum fi_type datatype)                                 = 0;
-    virtual void            w_fi_close(fid_t domain)                                                            = 0;
+    virtual int             w_fi_close(fid_t domain)                                                            = 0;
     virtual int             w_fi_fabric(struct fi_fabric_attr* attr, struct fid_fabric** fabric, void* context) = 0;
     virtual int
                 w_fi_domain(struct fid_fabric* fabric, struct fi_info* info, struct fid_domain** domain, void* context) = 0;