int mca_sharedfp_individual_file_close (mca_io_ompio_file_t *fh) { mca_sharedfp_individual_header_record* headnode = NULL; struct mca_sharedfp_base_data_t *sh; int err = OMPI_SUCCESS; if ( NULL == fh->f_sharedfp_data ){ if ( mca_sharedfp_individual_verbose ) { opal_output(ompi_sharedfp_base_framework.framework_output, "sharedfp_inidividual_file_close - shared file pointer structure not initialized\n"); } return OMPI_SUCCESS; } sh = fh->f_sharedfp_data; /* Merge data from individal files to final output file */ err = mca_sharedfp_individual_collaborate_data (sh); headnode = (mca_sharedfp_individual_header_record*)(sh->selected_module_data); if (headnode) { /*Close datafile*/ if (headnode->datafilehandle) { /*TODO: properly deal with returned error code*/ err = mca_common_ompio_file_close(headnode->datafilehandle); /* NOTE: No neeed to manually delete the file, ** the amode should have been set to delete on close */ } if(headnode->datafilename){ free(headnode->datafilename); } /*Close metadatafile*/ if (headnode->metadatafilehandle) { /*TODO: properly deal with returned error code*/ err = mca_common_ompio_file_close(headnode->metadatafilehandle); /* NOTE: No neeed to manually delete the file, ** the amode should have been set to delete on close */ } if(headnode->metadatafilename){ free(headnode->metadatafilename); } } /* Close the main file opened by this component*/ err = mca_common_ompio_file_close(sh->sharedfh); /*free shared file pointer data struct*/ free(sh); return err; }
int mca_sharedfp_individual_write_ordered (mca_io_ompio_file_t *fh, void *buf, int count, struct ompi_datatype_t *datatype, ompi_status_public_t *status) { int ret = OMPI_SUCCESS; int size = 0, rank = 0; int i = 0; size_t numofbytes = 0; size_t totalbytes = 0; OMPI_MPI_OFFSET_TYPE *offbuff=NULL; OMPI_MPI_OFFSET_TYPE global_offset = 0; OMPI_MPI_OFFSET_TYPE prev_offset = 0; OMPI_MPI_OFFSET_TYPE temp = 0, offset = 0; mca_sharedfp_individual_header_record *headnode = NULL; struct mca_sharedfp_base_data_t *sh = NULL; mca_sharedfp_base_module_t * shared_fp_base_module = NULL; if(fh->f_sharedfp_data==NULL){ if ( mca_sharedfp_individual_verbose ) { opal_output(ompi_sharedfp_base_framework.framework_output, "sharedfp_individual_write_ordered - opening the shared file pointer\n"); } shared_fp_base_module = fh->f_sharedfp; ret = shared_fp_base_module->sharedfp_file_open(fh->f_comm, fh->f_filename, fh->f_amode, fh->f_info, fh); if ( OMPI_SUCCESS != ret ) { opal_output(0,"sharedfp_individual_write_ordered - error opening the shared file pointer\n"); return ret; } } /*Retrieve the sharedfp data structures*/ sh = fh->f_sharedfp_data; rank = ompi_comm_rank ( sh->comm ); size = ompi_comm_size ( sh->comm ); /* Calculate the number of bytes of data that needs to be written*/ opal_datatype_type_size ( &datatype->super, &numofbytes); totalbytes = count * numofbytes; headnode = (mca_sharedfp_individual_header_record*)sh->selected_module_data; if ( NULL == headnode) { opal_output (0, "sharedfp_individual_write_ordered: headnode is NULL but file is open\n"); return OMPI_ERROR; } /* Data from all the metadata is combined and written to the main file */ ret = mca_sharedfp_individual_collaborate_data ( sh ); if ( OMPI_SUCCESS != ret) { return ret; } if ( 0 == rank ) { offbuff = (OMPI_MPI_OFFSET_TYPE *)malloc ( sizeof(OMPI_MPI_OFFSET_TYPE) * size); if (NULL == offbuff ) { return OMPI_ERR_OUT_OF_RESOURCE; } } /*collect the total bytes to be written*/ sh->comm->c_coll.coll_gather ( &totalbytes, 1, OMPI_OFFSET_DATATYPE, offbuff, 1, OMPI_OFFSET_DATATYPE, 0, sh->comm, sh->comm->c_coll.coll_gather_module ); if ( 0 == rank ) { prev_offset = offbuff[0]; offbuff[0] = sh->global_offset; for (i = 1; i < size ; i++){ temp = offbuff[i]; offbuff[i] = offbuff[i - 1] + prev_offset; prev_offset = temp; } for (i = 0; i < size; i++){ global_offset = offbuff[size - 1] + prev_offset; } } /* Scatter the results to the other processes */ ret = sh->comm->c_coll.coll_scatter ( offbuff, 1, OMPI_OFFSET_DATATYPE, &offset, 1, OMPI_OFFSET_DATATYPE, 0, sh->comm, sh->comm->c_coll.coll_scatter_module ); if ( OMPI_SUCCESS != ret ) { opal_output(0,"sharedfp_individual_write_ordered: Error in scattering offsets \n"); goto exit; } ret = sh->comm->c_coll.coll_bcast ( &global_offset, 1, OMPI_OFFSET_DATATYPE, 0, sh->comm, sh->comm->c_coll.coll_bcast_module ); if ( OMPI_SUCCESS != ret ) { opal_output(0,"sharedfp_individual_write_ordered: Error while bcasting global offset \n"); goto exit; } sh->global_offset = global_offset; /*use file_write_at_all to ensure the order*/ ret = ompio_io_ompio_file_write_at_all(sh->sharedfh,offset, buf,count,datatype,status); if ( OMPI_SUCCESS != ret ) { opal_output(0,"sharedfp_individual_write_ordered: Error while writing the datafile \n"); } exit: if ( NULL != offbuff ) { free ( offbuff); } return ret; }