Add check for empty inputs in getEnergyStats

Author: zhouselena

internal/cmd/perfcomp/energystatistics_test.go

@@ -82,6 +82,13 @@ func TestEnergyStatistics(t *testing.T) {
 		assert.NotEqual(t, nil, err)
 		assert.ErrorContains(t, err, "both inputs must have the same number of columns")
 
+		x.Reset()
+		y = &mat.Dense{}
+
+		_, _, _, err = getEnergyStatistics(x, y)
+		assert.NotEqual(t, nil, err)
+		assert.ErrorContains(t, err, "inputs cannot be empty")
+
 		x = mat.NewDense(2, 2, make([]float64, 4))
 		y = mat.NewDense(3, 2, make([]float64, 6))
 

internal/cmd/perfcomp/main.go

@@ -334,6 +334,9 @@ func getEnergyStatistics(x, y *mat.Dense) (float64, float64, float64, error) {
 	if xcols != ycols {
 		return 0, 0, 0, fmt.Errorf("both inputs must have the same number of columns")
 	}
+	if xrows == 0 || yrows == 0 {
+		return 0, 0, 0, fmt.Errorf("inputs cannot be empty")
+	}
 
 	xrowsf := float64(xrows)
 	yrowsf := float64(yrows)