Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Out variant for embedding_bag_4bit_rowwise_offsets #51324

Closed

Conversation

maratsubkhankulov
Copy link

Summary: Add out variant for embedding_bag_4bit_rowwise_offsets and add it to static runtime registry

Test Plan: TDB

Differential Revision: D26138322

@facebook-github-bot
Copy link
Contributor

facebook-github-bot commented Jan 29, 2021

💊 CI failures summary and remediations

As of commit 84d3173 (more details on the Dr. CI page):


  • 2/2 failures possibly* introduced in this PR
    • 2/2 non-CircleCI failure(s)

Extra GitHub checks: 1 failed


This comment was automatically generated by Dr. CI (expand for details).Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions to the (internal) Dr. CI Users group.

@facebook-github-bot facebook-github-bot added cla signed oncall: jit Add this issue/PR to JIT oncall triage queue fb-exported labels Jan 29, 2021
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D26138322

@codecov
Copy link

codecov bot commented Jan 29, 2021

Codecov Report

Merging #51324 (84d3173) into master (fe645fd) will increase coverage by 0.00%.
The diff coverage is 91.66%.

@@           Coverage Diff           @@
##           master   #51324   +/-   ##
=======================================
  Coverage   80.85%   80.86%           
=======================================
  Files        1938     1938           
  Lines      211184   211189    +5     
=======================================
+ Hits       170758   170770   +12     
+ Misses      40426    40419    -7     

Summary:
Pull Request resolved: pytorch#51324

Add out variant for embedding_bag_4bit_rowwise_offsets and add it to static runtime registry

Test Plan:
```
MKL_NUM_THREADS=1 OMP_NUM_THREADS=1 numactl -m 0 -C 1 buck-out/opt/gen/caffe2/caffe2/fb/predictor/ptvsc2_predictor_bench --scripted_model=$INLINE_CVR_DIR/210494966_0.predictor.disagg.remote_request_only_remote_cast.pt --pt_inputs=$INLINE_CVR_DIR/remote_ro_wrapped_input_data.pt --pt_enable_static_runtime=true --pt_cleanup_activations=true --pt_enable_out_variant=true --compare_results=true --iters=5000 --warmup_iters=5000 --num_threads=1 --do_profile=true
```

before:
```
0.789023 ms.    54.8408%. quantized::embedding_bag_4bit_rowwise_offsets (82 nodes)
```

after:
```
0.620817 ms.    49.7136%. quantized::embedding_bag_4bit_rowwise_offsets (82 nodes)
```

Differential Revision: D26138322

fbshipit-source-id: 0564ed4eaea702f6801f836364dde9419e71a5fb
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D26138322

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in 5b0a648.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cla signed fb-exported Merged oncall: jit Add this issue/PR to JIT oncall triage queue
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants