diff --git a/run-docker.sh b/run-docker.sh index 45937e93d7b5059cde5cd87afb3c966815ccc4ec..0cbdff79b117417e65fb7a8998ba199d08d0501e 100755 --- a/run-docker.sh +++ b/run-docker.sh @@ -97,5 +97,6 @@ docker run -t --rm --name $DOCKER_INST_NAME -it \ -v $VIVADO_PATH:$VIVADO_PATH \ -e VIVADO_PATH=$VIVADO_PATH \ -e FINN_INST_NAME=$DOCKER_INST_NAME \ +-e FINN_ROOT="/workspace/finn" \ -p $JUPYTER_PORT:$JUPYTER_PORT \ $DOCKER_TAG bash -c "$DOCKER_CMD" diff --git a/src/finn/util/basic.py b/src/finn/util/basic.py index afa77317dc3a704dcc51722d4c869c747f2f3851..ec5d31f637bdc89782ce3c8f76bec84ac9df4354 100644 --- a/src/finn/util/basic.py +++ b/src/finn/util/basic.py @@ -10,6 +10,19 @@ import onnx from finn.core.datatype import DataType +def get_finn_root(): + "Return the root directory that FINN is cloned into." + + try: + return os.environ["FINN_ROOT"] + except KeyError: + raise Exception( + """Environment variable FINN_ROOT must be set + correctly. Please ensure you have launched the Docker contaier correctly. + """ + ) + + def make_build_dir(prefix=""): """Creates a temporary folder with given prefix to be used as a build dir. Use this function instead of tempfile.mkdtemp to ensure any generated files