github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix shift op

Browse Source
This commit is contained in:
Rahul Batra
2023-04-10 15:05:45 -05:00
parent 3d71a6a034
commit c7ac25dc60
3 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
bin/triton-translate.cpp
View File

@@ -1,8 +1,5 @@
#include "triton/Dialect/Triton/IR/Dialect.h"
#include "triton/Dialect/TritonGPU/IR/Dialect.h"
#include "triton/Target/LLVMIR/LLVMIRTranslation.h"
#include "triton/Target/PTX/PTXTranslation.h"
#include "triton/Target/HSACO/HSACOTranslation.h"
#include "mlir/ExecutionEngine/ExecutionEngine.h"
#include "mlir/ExecutionEngine/OptUtils.h"
#include "mlir/IR/AsmState.h"
#include "mlir/IR/BuiltinOps.h"
#include "mlir/IR/Dialect.h"

1
python/src/triton.cc
View File

@@ -135,6 +135,7 @@ void init_triton_ir(py::module &&m) {
// we load LLVM because the frontend uses LLVM.undef for
// some placeholders
self.getOrLoadDialect<mlir::LLVM::LLVMDialect>();
self.getOrLoadDialect<mlir::tensor::TensorDialect>();
});
// .def(py::init([](){
// mlir::MLIRContext context;

9
scripts/amd/test.sh
View File

@@ -3,9 +3,6 @@
# clear
set -x
# export MLIR_ENABLE_DUMP=1
# export LLVM_IR_ENABLE_DUMP=1
# export AMDGCN_ENABLE_DUMP=1
# log dir
ROOT_DIR=$(pwd)
@@ -17,9 +14,11 @@ chmod -R 777 $LOG_DIR
sh scripts/amd/clean.sh
# UNIT_TEST="python/test/unit/language/test_core_amd.py"
UNIT_TEST="python/test/unit/language/test_core_amd.py::test_shift_op[int8-int8-<<]"
# UNIT_TEST="python/test/unit/language/test_core_amd.py::test_shift_op[int32-int32->>]"
# UNIT_TEST="python/test/unit/language/test_core.py::test_empty_kernel[float32]"
# UNIT_TEST="python/test/unit/language/test_core.py::test_bin_op"
UNIT_TEST="python/test/unit/language/test_core.py::test_bin_op[float32-float32-+]"
# UNIT_TEST="python/test/unit/language/test_core.py::test_bin_op[float32-float32-+]"
# UNIT_TEST="python/test/unit/language/test_core.py::test_bin_op[int8-float16-%]"
# UNIT_TEST="python/test/unit/language/test_elementwise.py"
@@ -39,3 +38,5 @@ if [ "$1" == "backtrace" ]; then
else
pytest --capture=tee-sys -rfs --verbose "$UNIT_TEST" 2>&1 | tee $LOG_DIR/unit_test.log
fi
bash scripts/amd/cache_print.sh 2>&1 |tee $LOG_DIR/cache.log