search

ROCm / composable_kernel

Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
https://rocm.docs.amd.com/projects/composable_kernel/en/latest/
Other
297 stars 113 forks source link

Fix in dropout lambda to avoid the compiling issue on some docker/compiler #1350

Closed qianfengz closed 3 months ago

qianfengz commented 3 months ago

Compiling issue is reported on both ck_tile and xformers c++ extension building after upgrading the docker to latest rocm/pytorch-nightly:latest

This PR is to regularize the using of C++ codes to strictly follow the C++17 specification