Conversation
| GraphDef* NGraphClusterManager::GetClusterGraph(int idx) { | ||
| std::lock_guard<std::mutex> guard(s_cluster_graphs_mutex); | ||
| return s_cluster_graphs[idx]; | ||
| return idx < s_cluster_graphs.size() ? s_cluster_graphs[idx] : nullptr; |
There was a problem hiding this comment.
Need this change since now we might have nothing stored in NGraphClusterManager, so need to safely avoid accessing out-of-bounds requests
| #include <utility> | ||
|
|
||
| #include "tensorflow/core/common_runtime/dma_helper.h" | ||
| #include "tensorflow/core/common_runtime/function.h" |
There was a problem hiding this comment.
Included for FunctionBody
| } | ||
| } | ||
| library { | ||
| function { |
There was a problem hiding this comment.
The library of graphdefs
| } | ||
| node { | ||
| name: "Sigmoid" | ||
| op: "IdentityN" |
There was a problem hiding this comment.
IdentityN due to Kanvi's change
| # Comparing with expected value | ||
| assert np.isclose(res1, exp).all() | ||
|
|
||
| @pytest.mark.skip(reason="Not passing through grappler") |
There was a problem hiding this comment.
@kanvi-nervana : I think this is a minimal repro of your squeeze-net issue
sayantan-nervana
requested review from
avijit-nervana,
kanvi-nervana and
sindhu-nervana
and removed request for
sindhu-nervana
test/python/test_flib.py
Outdated
| from __future__ import division | ||
| from __future__ import print_function | ||
|
|
||
| import pytest, pdb |
There was a problem hiding this comment.
remove pdb
test/python/common.py
Outdated
| os.environ['NGRAPH_TF_DISABLE_DEASSIGN_CLUSTERS'] = '1' | ||
| ngraph_bridge.enable() | ||
| with tf.Session(config=config) as sess: | ||
| with tf.Session(graph=graph, config=config) as sess: |
There was a problem hiding this comment.
Why do we need this change? Just Curious.
There was a problem hiding this comment.
when we write python code like a = tf.constant(...); b = tf.constant(...); c = a+b, this underlying graph is added to the default graph. When we read it from a pbtxt (like in this case), the graph is not added to default graph. So was passing the graph along to common.py to use during session construction.
But found a way to set the graph read from pbtxt as default graph (as_default)
| const FunctionLibraryDefinition flib = | ||
| *ctx->function_library()->GetFunctionLibraryDefinition(); | ||
| const FunctionDef* fdef = | ||
| flib.Find("Enc_" + to_string(m_ngraph_cluster) + "_native_segment"); |
There was a problem hiding this comment.
Can we use the node name (eg. ngraph_cluster_251) instead of this "Enc_" + to_string(m_ngraph_cluster)" ?
…naSystems/ngraph-tf into sarkars/flib_ngenc_compute
| std::getenv("NGRAPH_TF_DISABLE") != nullptr) { | ||
| std::getenv("NGRAPH_TF_DISABLE") != nullptr || | ||
| IsProcessedByNgraphPass(&graph)) { | ||
| NGRAPH_VLOG(0) << "NGTF_OPTIMIZER: Ngraph is disabled "; |
There was a problem hiding this comment.
Include the or condition in the error message.
shresthamalik
left a comment
shresthamalik
left a comment
There was a problem hiding this comment.
LGTM. Requested minor change.
https://github.com/NervanaSystems/ngraph-tf/pull/529/files#diff-4d55399542be3550bdfca644b849afa2R110
5 participants
No description provided.