mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
bcf8973654
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/38559 Test Plan: Make sure that `cmake path/to/CMakeLists.txt -DCMAKE_PREFIX_PATH=`python -c 'import torch;print(torch.utils.cmake_prefix_path)'` succeeds for CMake projects which depends on Torch package Differential Revision: D21644066 Pulled By: malfet fbshipit-source-id: c8e3cb2cbd7f969fadea6a3bccc41c4edb3ec546
15 lines
517 B
Python
15 lines
517 B
Python
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
|
|
from .throughput_benchmark import ThroughputBenchmark
|
|
|
|
import os.path as _osp
|
|
|
|
# Set the module for a given object for nicer printing
|
|
def set_module(obj, mod):
|
|
if not isinstance(mod, str):
|
|
raise TypeError("The mod argument should be a string")
|
|
obj.__module__ = mod
|
|
|
|
#: Path to folder containing CMake definitions for Torch package
|
|
cmake_prefix_path = _osp.join(_osp.dirname(_osp.dirname(__file__)), 'share', 'cmake')
|