Browse Source

llast

master
陈忠鑫 2 years ago
parent
commit
bbef4447f0
5 changed files with 7 additions and 5 deletions
  1. +3
    -2
      include/api/cell.h
  2. +1
    -1
      include/c_api/status_c.h
  3. +1
    -1
      mindspore/ccsrc/pipeline/pynative/base.h
  4. +1
    -0
      mindspore/python/mindspore/dataset/__init__.py
  5. +1
    -1
      mindspore/python/mindspore/dataset/core/datatypes.py

+ 3
- 2
include/api/cell.h View File

@@ -30,7 +30,7 @@ using Input = InputAndOutput;
using Output = InputAndOutput;


class MS_API CellBase {
class MS_API CellBase { // for CellCreator
public:
CellBase() = default;
virtual ~CellBase() = default;
@@ -72,7 +72,8 @@ class MS_API GraphCell final : public Cell<GraphCell> {
std::shared_ptr<GraphImpl> executor_;
};

class MS_API InputAndOutput {
class MS_API InputAndOutput { // for CellCreator
public:
InputAndOutput();
~InputAndOutput() = default;


+ 1
- 1
include/c_api/status_c.h View File

@@ -22,7 +22,7 @@
extern "C" {
#endif

enum MSCompCode {
enum MSCompCode { // 4 bytes
kMSCompCodeCore = 0x00000000u,
kMSCompCodeMD = 0x10000000u,
kMSCompCodeME = 0x20000000u,


+ 1
- 1
mindspore/ccsrc/pipeline/pynative/base.h View File

@@ -37,7 +37,7 @@ namespace py = pybind11;

enum RunOpArgsEnum { PY_PRIM = 0, PY_NAME, PY_INPUTS, PY_ARGS_NUM };

struct OpExecInfo {
struct OpExecInfo { // for runop build
bool is_nop_prim = false;
bool is_dynamic_shape = false;
bool is_mixed_precision_cast = false;


+ 1
- 0
mindspore/python/mindspore/dataset/__init__.py View File

@@ -40,6 +40,7 @@ Descriptions of common dataset terms are as follows:
- Iterator, the base class of dataset iterator for enumerating elements.
"""

# pylint: disable=unused-import
from .core import config
from .engine import *
from .engine.cache_client import DatasetCache


+ 1
- 1
mindspore/python/mindspore/dataset/core/datatypes.py View File

@@ -22,7 +22,7 @@ from mindspore._c_expression import typing
import mindspore.common.dtype as mstype


def mstype_to_detype(type_):
def mstype_to_detype(type_): # pylint: disable=too-many-return-statements
"""
Get de data type corresponding to mindspore dtype.



Loading…
Cancel
Save