Browse Source

Merge branch 'incubator-master' into sync_d9c74e0a

tags/v0.6.0-beta
jonyguo 5 years ago
parent
commit
78cc0d5d8d
6 changed files with 4 additions and 6 deletions
  1. +1
    -1
      graphengine
  2. +1
    -1
      mindspore/ccsrc/CMakeLists.txt
  3. +1
    -1
      mindspore/ccsrc/transform/op_declare.cc
  4. +0
    -1
      mindspore/nn/optim/optimizer.py
  5. +1
    -1
      tests/ut/cpp/CMakeLists.txt
  6. +0
    -1
      tests/ut/python/optimizer/test_optimize_with_parameter_groups.py

+ 1
- 1
graphengine

@@ -1 +1 @@
Subproject commit 579dcb75a990b533f9182733a6424f2bd66f0f23
Subproject commit c27e428e9698dd4f9b198008596676bc2d1b49aa

+ 1
- 1
mindspore/ccsrc/CMakeLists.txt View File

@@ -126,7 +126,7 @@ endif()

if (ENABLE_GE)
if(ENABLE_TRAIN)
target_link_libraries(mindspore ge_client_train hccl)
target_link_libraries(mindspore ge_runner hccl)
else ()
target_link_libraries(mindspore ge_client)
endif ()


+ 1
- 1
mindspore/ccsrc/transform/op_declare.cc View File

@@ -823,7 +823,7 @@ OUTPUT_MAP(RealDiv) = {{0, OUTPUT_DESC(y)}};
// Cast
INPUT_MAP(Cast) = {{1, INPUT_DESC(x)}};
INPUT_ATTR_MAP(Cast) = {{2, ATTR_DESC(dst_type, AnyTraits<GEType>())}};
ATTR_MAP(Cast) = {{"Truncate", ATTR_DESC(truncate, AnyTraits<bool>())}};
ATTR_MAP(Cast) = EMPTY_ATTR_MAP;
OUTPUT_MAP(Cast) = {{0, OUTPUT_DESC(y)}};

// Reciprocal


+ 0
- 1
mindspore/nn/optim/optimizer.py View File

@@ -279,7 +279,6 @@ class Optimizer(Cell):
current_dynamic_lr = self.gather(self.learning_rate[i], self.global_step, 0)
lr += (current_dynamic_lr,)
F.control_depend(lr, self.assignadd(self.global_step, 1))

else:
lr = self.learning_rate
if self.dynamic_lr:


+ 1
- 1
tests/ut/cpp/CMakeLists.txt View File

@@ -128,7 +128,7 @@ add_executable(ut_tests ${UT_SRCS} ${MINDSPORE_SRC_LIST} ${UT_SUTB_SRC_LIST})

if (ENABLE_GE)
if(ENABLE_TRAIN)
target_link_libraries(ut_tests PRIVATE graph ge_client_train)
target_link_libraries(ut_tests PRIVATE graph ge_runner)
else()
target_link_libraries(ut_tests PRIVATE graph ge_client)
endif()


+ 0
- 1
tests/ut/python/optimizer/test_optimize_with_parameter_groups.py View File

@@ -29,7 +29,6 @@ context.set_context(mode=context.GRAPH_MODE)

class LeNet5(nn.Cell):
""" LeNet5 definition """

def __init__(self):
super(LeNet5, self).__init__()
self.conv1 = nn.Conv2d(1, 6, 5, pad_mode='valid')


Loading…
Cancel
Save