fix warning

amarin16 · amarin16 · commit f398b6435a33 · 2024-09-30T12:44:16.000-07:00
diff --git a/onnxruntime/core/providers/cpu/nn/layer_norm_impl.cc b/onnxruntime/core/providers/cpu/nn/layer_norm_impl.cc
@@ -15,15 +15,15 @@ namespace onnxruntime {
 
 namespace {
 
-ORT_FORCEINLINE double* OnlyCreateBufferIfMLFloat16(double* p_output, int num_elems) {
+ORT_FORCEINLINE double* OnlyCreateBufferIfMLFloat16(double* p_output, int64_t num_elems) {
   return p_output;
 }
 
-ORT_FORCEINLINE float* OnlyCreateBufferIfMLFloat16(float* p_output, int num_elems) {
+ORT_FORCEINLINE float* OnlyCreateBufferIfMLFloat16(float* p_output, int64_t num_elems) {
   return p_output;
 }
 
-ORT_FORCEINLINE float* OnlyCreateBufferIfMLFloat16(MLFloat16* p_output, int num_elems) {
+ORT_FORCEINLINE float* OnlyCreateBufferIfMLFloat16(MLFloat16* p_output, int64_t num_elems) {
   return p_output == nullptr ? nullptr : new float[num_elems];
 }
 

Original file line number	Diff line number	Diff line change
`@@ -15,15 +15,15 @@ namespace onnxruntime {`
`15`	`15`
`16`	`16`	`namespace {`
`17`	`17`
`18`		`-ORT_FORCEINLINE double* OnlyCreateBufferIfMLFloat16(double* p_output, int num_elems) {`
	`18`	`+ORT_FORCEINLINE double* OnlyCreateBufferIfMLFloat16(double* p_output, int64_t num_elems) {`
`19`	`19`	`return p_output;`
`20`	`20`	`}`
`21`	`21`
`22`		`-ORT_FORCEINLINE float* OnlyCreateBufferIfMLFloat16(float* p_output, int num_elems) {`
	`22`	`+ORT_FORCEINLINE float* OnlyCreateBufferIfMLFloat16(float* p_output, int64_t num_elems) {`
`23`	`23`	`return p_output;`
`24`	`24`	`}`
`25`	`25`
`26`		`-ORT_FORCEINLINE float* OnlyCreateBufferIfMLFloat16(MLFloat16* p_output, int num_elems) {`
	`26`	`+ORT_FORCEINLINE float* OnlyCreateBufferIfMLFloat16(MLFloat16* p_output, int64_t num_elems) {`
`27`	`27`	`return p_output == nullptr ? nullptr : new float[num_elems];`
`28`	`28`	`}`
`29`	`29`