github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-11 22:10:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

add implementatino for GetGPULaunchConfig1D

Browse Source
This commit is contained in:
xs_courtesy
2024-03-14 11:13:40 +08:00
parent f366a5ea1f
commit 388e043930
7 changed files with 105 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions/csrc/common/target.h
View File

@@ -105,7 +105,7 @@ class Target {
static Target DefaultAscendTarget();
static Target DefaultCUDATarget() {
return Target(OS::Linux, Arch::CUDA, BitLen::k64);
return Target(OS::Linux, Arch::NVGPU, BitLen::k64);
}
friend std::ostream& operator<<(std::ostream& os, const Target& target);