Correctness fixes for compute_cached.

Stores to cache key now take into account that the index to Load/Store
is scaled by the sizeof the type.

The cache_tag support now properly records the expression.