return aten::gt to the list of fusable operations, add expected graphs #11150
Conversation
ngimel
requested review from
apaszke,
colesbury,
ezyang,
gchanan,
soumith and
zdevito
as code owners
test/test_jit.py
Outdated
| y = torch.randn(4, 4, dtype=torch.float, device='cuda') | ||
|
|
||
| ge = self.checkTrace(f, (x, y)) | ||
| self.assertExpectedGraph(ge.graph_for(x,y)) |
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
torch/csrc/jit/fusion_compiler.cpp
Outdated
| @@ -925,7 +925,14 @@ struct CUDAFusedKernel : public FusedKernel { | |||
| TORCH_NVRTC_CHECK(nvrtcGetPTXSize(program, &ptx_size)); | |||
| ptx.resize(ptx_size); | |||
| TORCH_NVRTC_CHECK(nvrtcGetPTX(program, ptx.data())); | |||
|
|
|||
| //need an active context for cuModuleLoadData | |||
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
|
I can also rebase just fine, and the split PR already has one merge conflict. Order not important. |
|
#10981 has been merged. :) |
There was a problem hiding this comment.
apaszke has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
Summary: This patch adds fused forward and backward for clamp to the jit. This is one item of #11118 . If it's OK, I'd be happy to also add some more of #11118 . The patch depends on #11150 , which I merged into master as a base. I'll rebase it when that or #10981 is merged. This is first serious jit patch, thank you, ngimel and the others for their guidance. All errors are my own. Pull Request resolved: #11574 Differential Revision: D9943090 Pulled By: apaszke fbshipit-source-id: c40954b8c28c374baab8d3bd89acc9250580dc67
Fixes one of #11118 issues.