Home
last modified time | relevance | path

Searched refs:gemm_cache_blocking (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/_inductor/codegen/
Dcpp_gemm_template.py384 if config.cpp.gemm_cache_blocking is not None:
385 blockings = [int(i) for i in config.cpp.gemm_cache_blocking.split(",")]
/external/pytorch/torch/_inductor/
Dconfig.py817 gemm_cache_blocking = os.environ.get("TORCHINDUCTOR_CPP_GEMM_CACHE_BLOCKING", None) variable in cpp