github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-09 21:09:18 +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

20
extensions/csrc/common/dev_info_mgr.h
View File

@@ -1,20 +0,0 @@
#pragma once
#include <memory>
#include "common/nvgpu_dev_info.h"
#include "target.h"
namespace colossalAI {
namespace common {
template <typename Ret>
class DevInfoMgr final {
public:
static std::unique_ptr<Ret> GetDevInfo(int device_num) const {
return std::make_unique<Ret>(device_num);
}
};
} // namespace common
} // namespace colossalAI

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);