-
Notifications
You must be signed in to change notification settings - Fork 75
[release/2.5] Fix the Build and Enable FA/ME UT on gfx950 for ROCM >= 6.5 #2311
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
[release/2.5] Fix the Build and Enable FA/ME UT on gfx950 for ROCM >= 6.5 #2311
Conversation
|
Jenkins build for 3c1c9e87f8386753925a7f67dbcc852071f8aa93 commit finished as ABORTED |
…:warp_size() (#2293) Fixes SWDEV-540240, SWDEV-540309, SWDEV-539989 ``` ... ``` 80cca70 created a static global variable that used `at::cuda::warp_size()` to initialize its value, which needs GPUs to be visible to query device properties. However, GPUs are not present on CPU-only build systems. Convert static variable into a static function, thus preventing static initialization. http://rocm-ci.amd.com/job/pyt_whl_docker_mainline/1461/artifact/build_artifacts.txt/*view*/ Ran microbenchmark to confirm basic functionality: ``` root@ubb4-rack-22:/var/lib/jenkins/pytorch-micro-benchmarking# python3 micro_benchmarking_pytorch.py --network resnet50 INFO: running forward and backward for warmup. INFO: running the benchmark.. OK: finished running benchmark.. --------------------SUMMARY-------------------------- Microbenchmark for network : resnet50 Num devices: 1 Dtype: FP32 Mini batch size [img] : 64 Time per mini-batch : 0.10158218145370483 Throughput [img/sec] : 630.0317544289736= ```
14abff4 to
3c1c9e8
Compare
|
Jenkins build for 3c1c9e87f8386753925a7f67dbcc852071f8aa93 commit finished as ABORTED |
|
Jenkins build for 3c1c9e87f8386753925a7f67dbcc852071f8aa93 commit finished as ABORTED |
|
Jenkins build for 3c1c9e87f8386753925a7f67dbcc852071f8aa93 commit finished as ABORTED |
|
Jenkins build for 3c1c9e87f8386753925a7f67dbcc852071f8aa93 commit finished as ABORTED |
|
Jenkins build for 3c1c9e87f8386753925a7f67dbcc852071f8aa93 commit finished as ABORTED |
Backport #2311 --------- Co-authored-by: Jeff Daily <jeff.daily@amd.com> Co-authored-by: Ethan Wee <Ethan.Wee@amd.com>
Tested build locally using:
registry-sc-harbor.amd.com/rocm-ci-images/compute-rocm-dkms-component-staging-hip:2277-ubunt6u-22.04