This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
!12660
fix gpu bceloss bug
From:
@huaweib
Reviewed-by: @zhoufeng54,@chujinjin Signed-off-by:
@chujinjin
tags/v1.2.0-rc1
mindspore-ci-bot
Gitee
4 years ago
parent
5589881ffe
4a3fcc086b
commit
a7bfc16aca
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-1
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/binary_cross_entropy_gpu_kernel.h
+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/binary_cross_entropy_gpu_kernel.h
View File
@@ -59,7 +59,7 @@ class BinaryCrossEntropyGpuKernel : public GpuKernel {
reduction_ = 2;
}
workspace_size_ = sizeof(T);
if (reduction_
=
= 0) {
if (reduction_
!
= 0) {
workspace_size_ *= input_size_;
}
InitSizeLists();
Write
Preview
Loading…
Cancel
Save