update mindspore/ccsrc/plugin/device/gpu/kernel/arrays/cast_gpu_kernel.h.

This commit is contained in:
王程浩 2022-07-04 15:27:15 +00:00 committed by Gitee
parent d824f2ae8e
commit 12a30c546a
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 2 deletions

View File

@ -98,8 +98,8 @@ class CastGpuKernelMod : public DeprecatedNativeGpuKernelMod {
}
private:
int input_size_;
int output_size_;
int64_t input_size_;
int64_t output_size_;
bool is_null_input_;
std::string kernel_name_;