wined3d: Set load_local_constsF in shader_set_function().

dlls/wined3d/shader.c

@@ -3390,6 +3390,8 @@ static HRESULT shader_set_function(struct wined3d_shader *shader,
         return WINED3DERR_INVALIDCALL;
     }
 
+    shader->load_local_constsF = shader->lconst_inf_or_nan;
+
     return WINED3D_OK;
 }
 
@@ -3787,8 +3789,6 @@ static HRESULT shader_init(struct wined3d_shader *shader, struct wined3d_device
         }
     }
 
-    shader->load_local_constsF = shader->lconst_inf_or_nan;
-
     return hr;
 }