[inductor] fix redis-related env vars in remote_cache.py (#127583)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/127583
Approved by: https://github.com/oulgen
diff --git a/torch/_inductor/remote_cache.py b/torch/_inductor/remote_cache.py
index 7c40f60..5bf3f50 100644
--- a/torch/_inductor/remote_cache.py
+++ b/torch/_inductor/remote_cache.py
@@ -27,17 +27,14 @@
def __init__(self, cache_id: str):
import redis
- self._cache_id = cache_id
- self._key_fmt = os.environ.get(
- "TORCHINDUCTOR_REDIS_KEY_FORMAT", "pt2:{cache_id}:{key}"
- )
+ self._key_fmt = f"pt2:{cache_id}:{{key}}"
self._redis = redis.Redis(
- host=os.environ.get("TRITON_REDIS_HOST", "localhost"),
- port=int(os.environ.get("TRITON_REDIS_PORT", 6379)),
+ host=os.environ.get("TORCHINDUCTOR_REDIS_HOST", "localhost"),
+ port=int(os.environ.get("TORCHINDUCTOR_REDIS_PORT", 6379)),
)
def _get_key(self, key: str) -> str:
- return self._key_fmt.format(cache_id=self._cache_id, key=key)
+ return self._key_fmt.format(key=key)
def get(self, key: str):
return self._redis.get(self._get_key(key))
