mirror of
https://github.com/wassname/ray.git
synced 2026-07-01 01:26:47 +08:00
Robert Nishihara
486cb64e3f
* Compile global scheduler with -Werror -Wall. * Compile plasma manager with -Werror -Wall. * Compile local scheduler with -Werror -Wall. * Compile common code with -Werror -Wall. * Signed/unsigned comparisons. * More signed/unsigned fixes. * More signed/unsigned fixes and added extern keyword. * Fix linting. * Don't check strict-aliasing because Python.h doesn't pass.
57 lines
1.2 KiB
C++
57 lines
1.2 KiB
C++
#ifndef COMMON_EXTENSION_H
|
|
#define COMMON_EXTENSION_H
|
|
|
|
#include <Python.h>
|
|
#include "marshal.h"
|
|
#include "structmember.h"
|
|
|
|
#include "common.h"
|
|
|
|
typedef uint8_t TaskSpec;
|
|
class TaskBuilder;
|
|
|
|
extern PyObject *CommonError;
|
|
|
|
// clang-format off
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
ObjectID object_id;
|
|
} PyObjectID;
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
int64_t size;
|
|
TaskSpec *spec;
|
|
} PyTask;
|
|
// clang-format on
|
|
|
|
extern PyTypeObject PyObjectIDType;
|
|
|
|
extern PyTypeObject PyTaskType;
|
|
|
|
/* Python module for pickling. */
|
|
extern PyObject *pickle_module;
|
|
extern PyObject *pickle_dumps;
|
|
extern PyObject *pickle_loads;
|
|
|
|
void init_pickle_module(void);
|
|
|
|
extern TaskBuilder *g_task_builder;
|
|
|
|
int PyStringToUniqueID(PyObject *object, ObjectID *object_id);
|
|
|
|
int PyObjectToUniqueID(PyObject *object, ObjectID *object_id);
|
|
|
|
PyObject *PyObjectID_make(ObjectID object_id);
|
|
|
|
PyObject *check_simple_value(PyObject *self, PyObject *args);
|
|
|
|
PyObject *PyTask_to_string(PyObject *, PyObject *args);
|
|
PyObject *PyTask_from_string(PyObject *, PyObject *args);
|
|
|
|
PyObject *compute_put_id(PyObject *self, PyObject *args);
|
|
|
|
PyObject *PyTask_make(TaskSpec *task_spec, int64_t task_size);
|
|
|
|
#endif /* COMMON_EXTENSION_H */
|