hetr_server.py 文件源码

def BuildTransformer(self, request, context):
        logger.debug("server: build_transformer")
        self.transformer_type = request.transformer_type[:3]
        if self.transformer_type not in ['gpu', 'cpu']:
            message = 'unknown transformer type {}'.format(self.transformer_type)
            return hetr_pb2.BuildTransformerReply(status=False, message=message)

        try:
            self.transformer = build_transformer(name=request.transformer_type, comm=self.comm)
            return hetr_pb2.BuildTransformerReply(status=True)
        except Exception:
            return hetr_pb2.BuildTransformerReply(status=False, message=traceback.format_exc())