Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Conversation

@DamianSzwichtenberg
Copy link
Member

Added several fixes and improvements for (segment|grouped)_matmul:

  • PyTorch stores shape information in int64_t variables, whereas MKL uses regular int. This means that tensor size can exceed the maximum value acceptable by MKL - in that case we need to use the ATen execution path.
  • config.h file was removed and added to the .gitignore list - it is automatically generated during the build from the config.h.in.
  • grain size in MKL sequential execution path was improved, resulting in an average 53% speedup.

@codecov-commenter
Copy link

Codecov Report

Merging #172 (09b0e61) into master (b78a67e) will increase coverage by 0.02%.
The diff coverage is 94.11%.

@@            Coverage Diff             @@
##           master     #172      +/-   ##
==========================================
+ Coverage   93.14%   93.16%   +0.02%     
==========================================
  Files          23       23              
  Lines         729      746      +17     
==========================================
+ Hits          679      695      +16     
- Misses         50       51       +1     
Impacted Files Coverage Δ
pyg_lib/csrc/ops/cpu/matmul_kernel.cpp 84.84% <94.11%> (+1.92%) ⬆️

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

Copy link
Member

@rusty1s rusty1s left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! This is cool.

PyTorch stores shape information in `int64_t` variables, whereas
MKL uses regular `int`. This means that tensor size can exceed
the maximum value acceptable by MKL - in that case we need to
use the ATen execution path
This file is generated automatically out of config.h.in
during build time
@DamianSzwichtenberg DamianSzwichtenberg merged commit 17d3e47 into pyg-team:master Jan 10, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants