dtk

8f7de847 · yuguo960516yuguo · f262efc9 · 8f7de847 · 8f7de847 · 8f7de847
Commit 8f7de847 authored Apr 25, 2023 by yuguo960516yuguo
20 changed files
--- a/oneflow/core/framework/random_generator_impl.hip.cpp
+++ b/oneflow/core/framework/random_generator_impl.hip.cpp
--- a/oneflow/core/hip/atomic.hip.h
+++ b/oneflow/core/hip/atomic.hip.h
--- a/oneflow/core/hip/elementwise.hip.h
+++ b/oneflow/core/hip/elementwise.hip.h
--- a/oneflow/core/hip/layer_norm.hip.h
+++ b/oneflow/core/hip/layer_norm.hip.h
--- a/oneflow/core/hip/softmax.hip.h
+++ b/oneflow/core/hip/softmax.hip.h
--- a/oneflow/core/hip/unique.hip.h
+++ b/oneflow/core/hip/unique.hip.h
--- a/oneflow/core/job/collective_boxing/nccl_executor_backend.hip.cpp
+++ b/oneflow/core/job/collective_boxing/nccl_executor_backend.hip.cpp
--- a/oneflow/core/job_rewriter/insert_nccl_logical_op_pass.cpp
+++ b/oneflow/core/job_rewriter/insert_nccl_logical_op_pass.cpp
@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 #include "oneflow/core/job/nd_sbp_util.h"
-#if defined(WITH_CUDA) || defined(ROCM)
+#if defined(WITH_CUDA) || defined(WITH_ROCM)
 #include "oneflow/core/framework/framework.h"
 #include "oneflow/core/framework/nd_sbp.h"
 #include "oneflow/core/framework/instructions_builder.h"

--- a/oneflow/core/job_rewriter/sequential_one_embedding_shuffle_ops_pass.cpp
+++ b/oneflow/core/job_rewriter/sequential_one_embedding_shuffle_ops_pass.cpp
--- a/oneflow/core/kernel/cuda_check_numerics_kernel_observer.hip.cpp
+++ b/oneflow/core/kernel/cuda_check_numerics_kernel_observer.hip.cpp
--- a/oneflow/core/kernel/kernel_util.hip.h
+++ b/oneflow/core/kernel/kernel_util.hip.h
--- a/oneflow/core/kernel/random_generator.hip.cpp
+++ b/oneflow/core/kernel/random_generator.hip.cpp
--- a/oneflow/core/kernel/util/numeric_limits.hip.h
+++ b/oneflow/core/kernel/util/numeric_limits.hip.h
--- a/oneflow/core/kernel/util/numerics.hip.h
+++ b/oneflow/core/kernel/util/numerics.hip.h
--- a/oneflow/core/ndarray/ndarray_apply_binary_core.hip.cpp
+++ b/oneflow/core/ndarray/ndarray_apply_binary_core.hip.cpp
--- a/oneflow/core/ndarray/ndarray_apply_broadcast_binary_core.hip.cpp
+++ b/oneflow/core/ndarray/ndarray_apply_broadcast_binary_core.hip.cpp
--- a/oneflow/core/ndarray/ndarray_apply_broadcast_unary_core.hip.cpp
+++ b/oneflow/core/ndarray/ndarray_apply_broadcast_unary_core.hip.cpp
--- a/oneflow/core/ndarray/ndarray_apply_unary_core.hip.cpp
+++ b/oneflow/core/ndarray/ndarray_apply_unary_core.hip.cpp
--- a/oneflow/core/ndarray/ndarray_assign_core.hip.cpp
+++ b/oneflow/core/ndarray/ndarray_assign_core.hip.cpp
--- a/oneflow/core/ndarray/ndarray_reduce_impl.hip.cpp
+++ b/oneflow/core/ndarray/ndarray_reduce_impl.hip.cpp