Searched refs:insert_device_ordinal (Results 1 – 5 of 5) sorted by relevance
191 def set_filename(filename: str, insert_device_ordinal: bool = False) -> None:198 …torch._C._cuda_tunableop_set_filename(filename, insert_device_ordinal) # type: ignore[attr-define…
205 void SetFilename(const std::string& filename, bool insert_device_ordinal=false);
565 void TuningContext::SetFilename(const std::string& filename, bool insert_device_ordinal) { in SetFilename() argument572 if (insert_device_ordinal) { in SetFilename()
119 | set_filename(filename: str, insert_device_ordinal: bool = False) -> None | |146 | void SetFilename(const std::string& filename, bool insert_device_ordinal=false); | |
1896 def _cuda_tunableop_set_filename(filename: str, insert_device_ordinal: Optional[_bool]) -> None: ...